Searched refs:grad_wrt_min (Results 1 – 1 of 1) sorted by relevance
242 Tensor* grad_wrt_min; in Compute() local244 context->allocate_output(1, scalar_shape, &grad_wrt_min)); in Compute()254 grad_wrt_min->scalar<float>(), grad_wrt_max->scalar<float>()); in Compute()386 Tensor* grad_wrt_min; in Compute() local388 context->allocate_output(1, min_max_shape, &grad_wrt_min)); in Compute()399 grad_wrt_min->vec<float>(), grad_wrt_max->vec<float>()); in Compute()