- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for unsorted_segment_sum (0.93 sec)
-
tensorflow/compiler/mlir/lite/schema/schema.fbs
BUCKETIZE = 147, RANDOM_UNIFORM = 148, MULTINOMIAL = 149, GELU = 150, DYNAMIC_UPDATE_SLICE = 151, RELU_0_TO_1 = 152, UNSORTED_SEGMENT_PROD = 153, UNSORTED_SEGMENT_MAX = 154, UNSORTED_SEGMENT_SUM = 155, ATAN2 = 156, UNSORTED_SEGMENT_MIN = 157, SIGN = 158, BITCAST = 159, BITWISE_XOR = 160, RIGHT_SHIFT = 161, // All Operators start with STABLEHLO_ prefixes are subject to change
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/cc/gradients/math_grad_test.cc
using ops::RealDiv; using ops::SegmentSum; using ops::SelectV2; using ops::SquaredDifference; using ops::Sub; using ops::Sum; using ops::UnsortedSegmentMax; using ops::UnsortedSegmentMin; using ops::UnsortedSegmentSum; using ops::Where3; // TODO(andydavis) Test gradient function against numeric gradients output. // TODO(andydavis) As more gradients are added move common test functions // to a testutil library.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 36K bytes - Viewed (0) -
tensorflow/cc/gradients/array_grad.cc
values = Reshape(scope, values, flat_values_shape); } indices = Reshape(scope, indices, indices_size); Output params_grad = UnsortedSegmentSum(scope, values, indices, gather_dim_size); if (batch_dims != 0) { // Put back the batch dimensions. params_grad = Reshape(scope, params_grad, params_shape); } return params_grad; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Oct 10 23:33:32 UTC 2023 - 31.7K bytes - Viewed (0)