- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for UnsortedSegmentSum (0.28 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/jit/mark_for_compilation_pass.cc
"UniformRequantize", "Unique", "UniqueV2", "UpperBound", "UnsortedSegmentMax", "UnsortedSegmentMin", "UnsortedSegmentProd", "UnsortedSegmentSum", "VarIsInitializedOp", "VariableShape", "Where", "While", "XlaAllReduce", "XlaBroadcastHelper", "XlaCallModule", "XlaConcatND",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 21 12:19:41 UTC 2024 - 85.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/ir/tfl_ops.td
} 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 summing together elements mapped to the same segment_id. I.e. `output[i]` is
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/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: mhlo.return [[ADD]] // CHECK-NEXT: (tensor<4x64xf32>, tensor<8x16xi32>, tensor<8x16x64xf32>) -> tensor<4x64xf32> // CHECK: return [[SCATTER]] %0 = "tf.UnsortedSegmentSum"(%data, %segment_ids, %num_segments) : (tensor<8x16x64xf32>, tensor<8x16xi32>, tensor<i32>) -> (tensor<4x64xf32>) func.return %0: tensor<4x64xf32> } // ----- // CHECK-LABEL: @unsorted_segment_prod
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/tensorflow/ir/tf_generated_ops.td
TF_DerivedOperandTypeAttr Tnumsegments = TF_DerivedOperandTypeAttr<2>; let hasVerifier = 1; } def TF_UnsortedSegmentSumOp : TF_Op<"UnsortedSegmentSum", [Pure]> { let summary = "Computes the sum along segments of a tensor."; let description = [{ Read [the section on segmentation](https://tensorflow.org/api_docs/python/tf/math#Segmentation)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 793K bytes - Viewed (0) -
RELEASE.md
* `tf.einsum` is supported with multiple unknown shapes. * `tf.unsortedsegmentprod` op is supported. * `tf.unsortedsegmentmax` op is supported. * `tf.unsortedsegmentsum` op is supported. * Updates to existing operations: * `tfl.scatter_nd` now supports I1 for the `update` arg. * Upgrade Flatbuffers v2.0.5 from v1.12.0 * `tf.keras`:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)