Searched refs:min_max_shape (Results 1 – 2 of 2) sorted by relevance
399 TensorShape min_max_shape({input.dim_size(input.dims() - 1)}); in Compute() local402 context->allocate_output(1, min_max_shape, &grad_wrt_min)); in Compute()406 context->allocate_output(2, min_max_shape, &grad_wrt_max)); in Compute()
175 TensorShape min_max_shape(input_min_tensor.shape()); in Compute() local178 ctx->allocate_output(1, min_max_shape, &input_min_backprop)); in Compute()182 ctx->allocate_output(2, min_max_shape, &input_max_backprop)); in Compute()