Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for _TPUDeviceOrdinalPlaceholder (0.19 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/replicate_to_island.mlir

            %1 = "tf._TPUDeviceOrdinalPlaceholder"() {logical_core = 0} : () -> tensor<i64>
            %2 = "tf._TPUDeviceOrdinalPlaceholder"() {logical_core = 1} : () -> tensor<i64>
            tf_device.return
          }
          tf_executor.yield
        }
        tf_executor.fetch
      }
      func.return
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 15.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/replicate_invariant_op_hoisting.mlir

    func.func @not_invariant_ordinal_placeholder(%arg0: tensor<*xf32>, %arg1: tensor<*xf32>) {
      // CHECK: tf_device.replicate
      // CHECK:   tf._TPUDeviceOrdinalPlaceholder
      %0:2 = tf_device.replicate([%arg0, %arg1] as %ri: tensor<*xf32>) {n = 2: i32} {
        %1 = "tf._TPUDeviceOrdinalPlaceholder"() {logical_core = 0} : () -> tensor<i64>
        tf_device.return %1 : tensor<i64>
      }
      func.return
    }
    
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 11.9K bytes
    - Viewed (0)
Back to top