Searched refs:mean_gradients_t (Results 1 – 1 of 1) sorted by relevance
260 const Tensor* mean_gradients_t; in Compute() local262 context->input("mean_gradients", &mean_gradients_t)); in Compute()280 const float gradients_mean = mean_gradients_t->flat<float>()(0); in Compute()