Search Options

Results per page
Sort
Preferred Languages
Advance

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

  1. tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir

        // CHECK-NEXT:     "tf_device.launch"
        // CHECK-DAG:        %[[PROGRAM_OUTPUT:[a-z_0-9]*]] = "tf._XlaCompileMlirPlaceholderProgramKey"
        // CHECK-DAG:        %[[DEVICE_ORDINAL:[a-z_0-9]+]] = "tf._TPUDeviceOrdinalPlaceholder"
        // CHECK:            %[[RECV_OUTPUT:[0-9]*]] = "tf._XlaRecvAtHostV2"(%[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]])
        // CHECK-SAME:       key = "host_compute_channel_0_args"
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 129.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir

        // expected-remark@above {{ID: 4}}
      }
      func.return
      // expected-remark@above {{ID: 6}}
      // expected-remark@above {{Sinks: {}}}
    }
    
    // -----
    
    // Tests that `_TPUDeviceOrdinalPlaceholder` is side-effect-free.
    func.func @device_ordinal_placeholder_side_effect_free(
      // expected-remark@above {{ID: 7}}
      ) {
      tf_executor.graph {
        // expected-remark@above {{ID: 5}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 20 04:39:18 UTC 2023
    - 129.7K bytes
    - Viewed (0)
Back to top