- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for UnsortedSegmentProd (0.41 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/einsum.mlir
// CHECK-DAG: %[[cst_1:.*]] = arith.constant dense<[-1, 36, 32]> : tensor<3xi64> // CHECK: %[[v0:.*]] = "tf.Shape"(%arg0) : (tensor<?x36x?xf32>) -> tensor<3xi32> // CHECK: %[[v1:.*]] = "tf.UnsortedSegmentProd"(%0, %[[cst]], %cst_0) : (tensor<3xi32>, tensor<3xi32>, tensor<i32>) -> tensor<4xi32> // CHECK: %[[v2:.*]] = "tf.Reshape"(%arg0, %1) : (tensor<?x36x?xf32>, tensor<4xi32>) -> tensor<?x36x1x?xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jan 05 18:35:42 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/einsum.cc
} // Creates ReshapeOp with runtime calcuation of required shape to support // dynamic shapes. The shape is calculated by Shape and UnsortedSegmentProd op. // `reshape_segids` and `num_reshape_segids` for UnsortedSegmentProd is // calculated in `reshapeForBatchMatmul`. TF::ReshapeOp createReshapeOpForDynamic(Value value, ArrayRef<int64_t> shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK-DAG: %cst_2 = "tf.Const"() <{value = dense<1> : tensor<i32>}> : () -> tensor<i32> // CHECK: %2 = "tf.UnsortedSegmentProd"(%1, %cst_0, %cst_2) : (tensor<3xi32>, tensor<3xi32>, tensor<i32>) -> tensor<1xi32> // CHECK: %3 = "tf.UnsortedSegmentProd"(%1, %cst_1, %cst_2) : (tensor<3xi32>, tensor<3xi32>, tensor<i32>) -> tensor<1xi32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/jit/mark_for_compilation_pass.cc
"UniformQuantizedDot", "UniformRequantize", "Unique", "UniqueV2", "UpperBound", "UnsortedSegmentMax", "UnsortedSegmentMin", "UnsortedSegmentProd", "UnsortedSegmentSum", "VarIsInitializedOp", "VariableShape", "Where", "While", "XlaAllReduce", "XlaBroadcastHelper", "XlaCallModule",
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/tests/legalize-tf.mlir
} func.func @unsorted_segment_prod(%arg0: tensor<8xf32>, %arg1: tensor<8xi32>) -> tensor<8xf32> { %num_segments = "tf.Const"() {value = dense<8> : tensor<i32>} : () -> tensor<i32> %0 = "tf.UnsortedSegmentProd"(%arg0, %arg1, %num_segments) : (tensor<8xf32>, tensor<8xi32>, tensor<i32>) -> tensor<8xf32> func.return %0 : tensor<8xf32> // CHECK-LABEL: unsorted_segment_prod
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/ir/tfl_ops.td
} def TFL_UnsortedSegmentProdOp: TFL_Op<"unsorted_segment_prod", [ Pure, PredOpTrait<"input and output must have same element type", TFL_TCresVTEtIsSameAsOp<0, 0>>]> { let summary = "UnsortedSegmentProd operator"; let description = [{ Computes the product along segments of a tensor. }]; let arguments = (ins TFL_TensorOf<[F32, I32]>:$input, TFL_I32Tensor:$segment_ids,
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/tensorflow/tests/tf-ops.mlir
// expected-error @+1 {{requires segment ids shape to be a prefix of data shape, but dimension #1 differs: 9 vs. 10}} %0 = "tf.UnsortedSegmentProd"(%data, %segment_ids, %num_segments) : (tensor<7x10x8xf32>, tensor<7x9xi32>, tensor<i32>) -> (tensor<?x8xf32>) func.return } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 23 14:40:35 UTC 2023 - 236.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/tests/legalize-tf.mlir
// CHECK: mhlo.return [[MUL]] // CHECK-NEXT: (tensor<4x64xf32>, tensor<?x16xi32>, tensor<8x?x64xf32>) -> tensor<4x?xf32> // CHECK: return [[SCATTER]] %0 = "tf.UnsortedSegmentProd"(%data, %segment_ids, %num_segments) : (tensor<8x?x64xf32>, tensor<?x16xi32>, tensor<i32>) -> (tensor<4x?xf32>) func.return %0: tensor<4x?xf32> } // ----- // CHECK-LABEL: @unsorted_segment_min
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_UnsortedSegmentProdOp : TF_Op<"UnsortedSegmentProd", [Pure]> { let summary = "Computes the product 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.lite`: * New operations supported: * tflite SelectV2 now supports 5D. * `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:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 23:24:08 UTC 2024 - 730.3K bytes - Viewed (0)