- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TPUCompileMlirAndExecute (0.21 sec)
-
tensorflow/compiler/mlir/tfrt/tests/mlrt/tpu_conversions.mlir
%1 = "tf.TPUCompileMlirAndExecute"(%0) {metadata = "metadata", mlir_module = "mlir_module", operandSegmentSizes = array<i32: 1, 0>, producer_name = "producer_name"} : (tensor<i32>) -> tensor<i32> func.return %1 : tensor<i32> } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Oct 04 21:25:31 UTC 2023 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir
// CHECK-NOT: tf.TPUCompileSucceededAssert // CHECK-NOT: tf.TPUExecuteOp // CHECK-NEXT: %0 = "tf.ReadVariableOp"(%arg1)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.td
TF_I32OrI64Tensor:$static_shape ); let hasVerifier = 1; let results = (outs TF_Tensor:$output ); } def TF_TPUCompileMlirAndExecuteOp : TF_Op<"TPUCompileMlirAndExecute", [AttrSizedOperandSegments]> { let summary = "Op that compiles a computation in MLIR into a TPU program, and loads and executes it on a TPU device."; let description = [{
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 90.5K bytes - Viewed (0)