Searched refs:grad_values_t (Results 1 – 1 of 1) sorted by relevance
205 const Tensor* grad_values_t; in Compute() local208 OP_REQUIRES_OK(context, context->input("grad_values", &grad_values_t)); in Compute()218 const auto grad_values = grad_values_t->vec<T>(); in Compute()221 const int64 N_full = grad_values_t->shape().dim_size(0); in Compute()