Searched defs:grad_val (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dsparse_conditional_accumulator.h162 const Tensor* grad_val = std::get<1>(*grad); variable
176 ctx->allocate_persistent(dtype_, grad_val->shape(), accum_val_persistent_,
180 grad_val->flat<T>();
198 const Tensor* grad_val = std::get<1>(*grad); variable
258 TensorShape sum_shape = grad_val->shape();
266 auto grad_flat = grad_val->flat_outer_dims<T>();

Completed in 172 milliseconds