Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for 65536xi64 (0.07 sec)

  2. github.com/tensorflow/tensorflow

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

      // CHECK: return
  func.func @copy_and_send(%arg0: tensor<65536xi64>, %arg1: tensor<1x!tf_type.string>, %arg2: tensor<65536xi32>) {
    "tf_device.launch"() ({
	%7088 = "tf.TPUCopyWithDynamicShape"(%arg2, %arg2) {operandSegmentSizes = array<i32: 1, 1>} : (tensor<65536xi32>, tensor<65536xi32>) -> tensor<65536xi64>
	"tf._XlaSendFromHostV2"(%arg1, %7088) {key = "foo"} : (tensor<1x!tf_type.string>, tensor<65536xi64>) -> ()
	tf_device.return
    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