Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for _XlaSendFromHostV2 (0.19 sec)

  2. github.com/tensorflow/tensorflow

    tensorflow/compiler/mlir/tensorflow/tests/extract_tpu_copy_with_dynamic_shape_op.mlir 

        return %0#0, %0#1: tensor<2048xi32>, tensor<2048xi32>
  }

  // CHECK-LABEL: func @copy_and_send

  // CHECK: "tf_device.launch"
  // CHECK: "tf.TPUCopyWithDynamicShape"
  // CHECK: "tf._XlaSendFromHostV2
  // CHECK: tf_device.return
  // CHECK-NOT: launch
  // CHECK: return
  func.func @copy_and_send(%arg0: tensor<65536xi64>, %arg1: tensor<1x!tf_type.string>, %arg2: tensor<65536xi32>) {
    "tf_device.launch"() ({
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Oct 31 08:59:10 UTC 2023
    - 4.9K bytes
    - Viewed (0)
Back to top