- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for BatchMatMulV3 (0.16 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
// CHECK: %[[VAL_3:.*]] = "tf.Reshape"(%[[VAL_1]], %[[VAL_2]]) : (tensor<256xf32>, tensor<2xi64>) -> tensor<256x1xf32> // CHECK: %[[VAL_4:.*]] = "tf.BatchMatMulV3"(%[[VAL_0]], %[[VAL_3]]) <{adj_x = false, adj_y = false, grad_x = false, grad_y = false}> : (tensor<1x256xf32>, tensor<256x1xf32>) -> tensor<1x1xf32> // CHECK: %[[VAL_5:.*]] = arith.constant dense<1> : tensor<1xi64>
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/mlir/tf2xla/transforms/legalize_tf.cc
// Set benefit of this pattern to zero to prefer the fallback pattern when // available and applicable. That pattern avoids broadcast on operands and is // therefore faster. // // Native legalization for BatchMatMulV3 needs to be added as well. explicit ConvertBatchMatMulV2Op(MLIRContext *context) : OpRewritePattern<TF::BatchMatMulV2Op>(context, /*benefit=*/0) {} LogicalResult matchAndRewrite(TF::BatchMatMulV2Op op,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (0)