- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 14 for unsorted_segment_sum (0.4 sec)
-
tensorflow/compiler/mlir/lite/tests/legalize-tf.mlir
} // ----- func.func @unsorted_segment_sum(%arg0: tensor<8xf32>, %arg1: tensor<8xi32>) -> tensor<8xf32> { %num_segments = "tf.Const"() {value = dense<8> : tensor<i32>} : () -> tensor<i32> %0 = "tf.UnsortedSegmentSum"(%arg0, %arg1, %num_segments) : (tensor<8xf32>, tensor<8xi32>, tensor<i32>) -> tensor<8xf32> func.return %0 : tensor<8xf32> // CHECK-LABEL: unsorted_segment_sum
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 05 01:54:33 UTC 2024 - 153.4K bytes - Viewed (0) -
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/compiler/mlir/lite/tests/ops.mlir
// ----- // CHECK-LABEL: testUnsortedSegmentSum func.func @testUnsortedSegmentSum(%arg0: tensor<8xf32>, %arg1: tensor<8xi32>, %arg2: tensor<i32>) -> tensor<8xf32> { // CHECK: "tfl.unsorted_segment_sum"(%arg0, %arg1, %arg2) %0 = "tfl.unsorted_segment_sum"(%arg0, %arg1, %arg2) : (tensor<8xf32>, tensor<8xi32>, tensor<i32>) -> tensor<8xf32> func.return %0 : tensor<8xf32> // CHECK: return %0 : tensor<8xf32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 189.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
//===----------------------------------------------------------------------===// // ----- // CHECK-LABEL: @unsorted_segment_sum // CHECK-SAME: [[DATA:%.*]]: tensor<8x16x64xf32> // CHECK-SAME: [[SI:%.*]]: tensor<8x16xi32> func.func @unsorted_segment_sum(%data: tensor<8x16x64xf32>, %segment_ids : tensor<8x16xi32>) -> (tensor<4x64xf32>) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
); let results = (outs TFL_TensorOf<[F32, I32]>:$output); } def TFL_UnsortedSegmentSumOp: TFL_Op<"unsorted_segment_sum", [ Pure, PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>]> { let summary = "UnsortedSegmentSum operator"; let description = [{ From a tensor segmentation, computes the `output` resulting from
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 19:09:08 UTC 2024 - 186K bytes - Viewed (0) -
tensorflow/compiler/jit/tests/opens2s_gnmt_mixed_precision.golden_summary
cluster 29 size 9 Add 1 Mul 2 RealDiv 2 Sqrt 2 Sub 2 cluster 30 size 9 Add 1 Mul 2 RealDiv 2 Sqrt 2 Sub 2 cluster 31 size 4 Mul 3 UnsortedSegmentSum 1 cluster 32 size 4 Mul 3
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 06 10:38:14 UTC 2023 - 5K bytes - Viewed (0) -
RELEASE.md
* (since v2.7) Stateful ops used in `tf.data.Dataset` * (since v2.7) `tf.convert_to_tensor` when fed with (sparse) `tf.IndexedSlices` (because it uses `tf.math.unsorted_segment_sum`) * (since v2.7) `tf.gather` backprop (because `tf.convert_to_tensor` reduces `tf.gather`'s (sparse) `tf.IndexedSlices` gradients into its dense `params` input)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td
<div style="width:70%; margin:auto; margin-bottom:10px; margin-top:20px;"> <img style="width:100%" src="https://www.tensorflow.org/images/UnsortedSegmentSum.png" alt> </div> >>> c = [[1,2,3,4], [5,6,7,8], [4,3,2,1]] >>> tf.math.unsorted_segment_sum(c, [0, 1, 0], num_segments=2).numpy() array([[5, 5, 5, 5], [5, 6, 7, 8]], dtype=int32) }]; let arguments = (ins
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
tensorflow/cc/gradients/math_grad.cc
std::vector<Output>* grad_outputs) { if (op.num_inputs() != 3) { return errors::InvalidArgument("UnsortedSegmentSum requires 3 arguments"); } if (grad_inputs.size() != 1) { return errors::InvalidArgument( "UnsortedSegmentSum grad requires 1 grad input"); } auto zero_clipped_indices = GetZeroClippedIndices(scope, op.input(1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Aug 25 18:20:20 UTC 2023 - 50.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)