- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for UnsortedSegmentProd (0.19 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)