Search Options

Results per page
Sort
Preferred Languages
Advance

Results 41 - 43 of 43 for _device_ordinal (0.14 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/g3doc/_includes/tf_passes.md 

          %1 = "tf._XlaCompileMlirPlaceholderProgramKey"() : () -> tensor<3x!tf_type.string>
      %2 = "tf._XlaRecvAtHost"(%1) {device_ordinal = 0 : i64, key = "host_compute_channel_0_0_args"} : (tensor<3x!tf_type.string>) -> tensor<f32>
      %3 = "tf.Identity"(%2) : (tensor<f32>) -> tensor<f32>
      "tf._XlaSendFromHost"(%3, %1) {device_ordinal = 0 : i64, key = "host_compute_channel_0_0_retvals"} : (tensor<f32>, tensor<3x!tf_type.string>) -> ()
      tf_device.return
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Aug 02 02:26:39 UTC 2023
    - 96.4K bytes
    - Viewed (0)
  3. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td 

    which to execute computations, such as `TPUPartitionedCall`.

device_ordinal: An integer in [0, NumTPUCoresVisiblePerHost].
  }];

  let arguments = (ins);

  let results = (outs
    TF_Int32Tensor:$device_ordinal
  );
}

def TF_TakeDatasetOp : TF_Op<"TakeDataset", [Pure]> {
  let summary = [{
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
  4. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir 

          %1 = "tf_device.launch"() ({
        %2 = "tf._XlaCompileMlirPlaceholderProgramKey"() : () -> tensor<3x!tf_type.string>
        %3 = "tf._XlaRecvAtHost"(%2) {_xla_has_host_transfer = true, device_ordinal = 0 : i64, key = "host_compute_channel_0_args"} : (tensor<3x!tf_type.string>) -> tensor<?xi32>
        %4 = "tf.B"(%3) : (tensor<?xi32>) -> tensor<?x!tf_type.string>
        tf_device.return %4 : tensor<?x!tf_type.string>
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 02 22:03:30 UTC 2024
    - 172.9K bytes
    - Viewed (0)
Back to top