- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 12 for _XlaSendFromHost (0.53 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/tpu_tail_with_tobool_op.mlir
// host-device communication. // CHECK: tf._TPUCompileMlir // CHECK-SAME: tf.Rank // CHECK-NOT: tf._XlaHostComputeMlir // CHECK-NOT: tf._XlaRecvAtHost // CHECK-NOT: tf._XlaSendFromHost
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 13 21:23:47 UTC 2024 - 2.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/tests/fuse_tpu_compile_and_execute_ops.mlir
// CHECK-NEXT: "tf._XlaSendFromHost"(%arg0, %0, [[key]]) <{device_ordinal = 0 : i64, key = "host_compute_channel_0_retvals"}> {_xla_has_host_transfer = true, device = "/job:localhost/replica:0/task:0/device:CPU:0"} : (tensor<*xi32>, tensor<*xi32>, tensor<3x!tf_type.string>) ->...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Oct 30 06:52:55 UTC 2023 - 13.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir
// CHECK-SAME: key = "host_compute_channel_send" // CHECK-SAME: -> (tensor<*xf32>, tensor<3x?xf64>) // CHECK: %[[ID_OUTPUT:[0-9]*]] = "tf.Identity"(%[[RECV_OUTPUT]]#0) // CHECK: "tf._XlaSendFromHost" // CHECK-SAME: [[ID_OUTPUT]] // CHECK-SAME: key = "host_compute_channel_recv"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 18:46:36 UTC 2024 - 9.2K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
&call_fbody)); // Verify we have _XlaRecvAtHost and _XlaSendFromHost nodes. bool has_recv = false, has_send = false; for (Node *n : call_fbody->graph->nodes()) { if (n->type_string() == "_XlaRecvAtHost") { has_recv = true; } else if (n->type_string() == "_XlaSendFromHost") { has_send = true; } } EXPECT_TRUE(has_recv);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 41K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/extract_outside_compilation.mlir
// CHECK-NOT: "tf._TPUDeviceOrdinalPlaceholder" // CHECK: %[[B_OUTPUT:[0-9]*]] = "tf.B"() // CHECK: "tf._XlaSendFromHost"(%[[B_OUTPUT]], %[[PROGRAM_OUTPUT]]) // CHECK-SAME: device_ordinal = 0 // CHECK-SAME: device_type = "TPU" // CHECK-SAME: key = "host_compute_channel_0_retvals"
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/jit/extract_outside_compilation_pass.cc
for (Node* n : g->nodes()) { if (n->type_string() == "_XlaSendFromHost") { send_from_host = n; break; } } if (!send_from_host) { return errors::Internal("Shape inference graph ", shape_inference_graph_name, " does not have _XlaSendFromHost node."); } // See if the SendFromHost node exists in `host_graph`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/clustering_passes.td
%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 }) {device = "/job:worker/replica:0/task:0/device:CPU:0"} : () -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 19.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
} Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) { 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,
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/jit/encapsulate_subgraphs_pass_test.cc
string name = absl::StrCat("outside_compilation_", cluster, "_", new_func_name, "_", oc_cluster, "_send"); NodeBuilder node_builder(opts.WithName(name).GetNameForOp("_XlaSendFromHost"), "_XlaSendFromHost", opts.op_registry()); node_builder.Input(inputs); node_builder.Input(std::move(key_input)); std::vector<DataType> dtypes; for (const auto& node : inputs) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
// CHECK: %5 = "tf.Shape"(%4) {_xla_outside_compilation = "0"} : (tensor<1120x8xi32>) -> tensor<2xi32> // CHECK: "tf._XlaSendFromHost"(%3#1, %3#2, %4, %5, %0) <{device_ordinal = 0 : i64, device_type = "TPU", key = "host_compute_channel_0_retvals"}> : (tensor<1x1120x8xi32>, tensor<1x1120x8xi32>, tensor<1120x8xi32>, tensor<2xi32>, tensor<3x!tf_type.string>) -> ()
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)