- Sort Score
- Result 10 results
- Languages All
Results 1 - 1 of 1 for reduce_max (0.32 sec)
-
tensorflow/cc/gradients/math_grad.cc
auto x_shape = Shape(scope, x); auto output_shape = Shape(scope, op.output(0)); // Reduce away broadcasted leading dims. auto reduce_x = internal::BroadcastGradientArgs(scope, x_shape, output_shape); auto gx_sum = ReduceSum(scope, gx, /*axis=*/reduce_x.r0, ReduceSum::KeepDims(true)); auto gx_sum_reshape = Reshape(scope, gx_sum, x_shape); auto gy = SelectV2(scope, c, zeros, grad_inputs[0]);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.7K bytes - Viewed (0)