Searched refs:grad_wrt_max (Results 1 – 1 of 1) sorted by relevance
246 Tensor* grad_wrt_max; in Compute() local248 context->allocate_output(2, scalar_shape, &grad_wrt_max)); in Compute()254 grad_wrt_min->scalar<float>(), grad_wrt_max->scalar<float>()); in Compute()390 Tensor* grad_wrt_max; in Compute() local392 context->allocate_output(2, min_max_shape, &grad_wrt_max)); in Compute()399 grad_wrt_min->vec<float>(), grad_wrt_max->vec<float>()); in Compute()