Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for _XlaSendFromHostV2 (0.24 sec)

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

        // CHECK:            "tf._XlaSendFromHostV2"(%[[C_OUTPUT]]#0, %[[C_OUTPUT]]#1, %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]])
        // CHECK:            %[[RECV_OUTPUT:[0-9]*]]:2 = "tf._XlaRecvAtHostV2"
        // CHECK:            %[[E_OUTPUT:[0-9]*]] = "tf.E"(%[[RECV_OUTPUT]]#0, %[[RECV_OUTPUT]]#1)
        // CHECK:            "tf._XlaSendFromHostV2"(%[[E_OUTPUT]], %[[PROGRAM_OUTPUT]], %[[DEVICE_ORDINAL]])
    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/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)
  3. tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir

            // expected-remark@above {{ID: 0}}
            "tf._XlaSendFromHostV2"(%const, %arg0, %arg1) {_xla_has_host_transfer = true, key = "host_compute_channel_0_retvals"} : (tensor<f32>, tensor<!tf_type.string>, tensor<i64>) -> ()
            // expected-remark@above {{ID: 1}}
            // expected-remark@above {{Successors: {2}}}
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Dec 20 04:39:18 UTC 2023
    - 129.7K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc

      op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true));
      return op;
    }
    
    // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal
    // is present, a tf._XlaSendFromHostV2 op is created instead.
    Operation* CreateSendFromHostOp(OpBuilder& builder, Location loc,
                                    ValueRange inputs, Value compilation_key,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Apr 30 21:25:12 UTC 2024
    - 68.3K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/ir/tf_generated_ops.td

      );
    
      let results = (outs);
    
      TF_DerivedOperandTypeListAttr Tinputs = TF_DerivedOperandTypeListAttr<0>;
    }
    
    def TF__XlaSendFromHostV2Op : TF_Op<"_XlaSendFromHostV2", [DeclareOpInterfaceMethods<TF_GetResourceInstanceInterface>, TF_SendSideEffect]> {
      let summary = [{
    A placeholder op to send values to a running XLA computation with support for a runtime device ordinal.
      }];
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 23:24:08 UTC 2024
    - 793K bytes
    - Viewed (0)
Back to top