- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for _XlaSendFromHostV2 (0.24 sec)
-
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) -
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) -
tensorflow/compiler/mlir/tensorflow/tests/side-effect-analysis-test.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Dec 20 04:39:18 UTC 2023 - 129.7K bytes - Viewed (0) -
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) -
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)