Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for SomeTpuOp (0.08 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/tests/executor_tpuv1_outline_island/executor_tpuv1_outline_tpu_island.mlir 

    // CHECK-NEXT: tf.SomeTpuOp
// CHECK-NEXT: tf.SomeOtherTpuOp

// -----

// Test single-core TPU case with single-op cluster.

// CHECK-LABEL: @func4
func.func @func4() {
  tf_executor.graph {
// CHECK: island
// CHECK: PartitionedCall
// CHECK-SAME: @_tpu_v1_compat_outlined::@_tpu_v1_compat_outlined_func0
    %1 = tf_executor.island {
     "tf.SomeTpuOp"() {_xla_compile_device_type = "TPU"} : () -> ()
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 04 03:54:58 UTC 2022
    - 4.9K bytes
    - Viewed (0)
Back to top