- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for _xla_token_arg_node (0.25 sec)
-
tensorflow/compiler/mlir/tensorflow/tests/prepare_tpu_computation_for_tf_export.mlir
func.return } // CHECK-LABEL: @CommunicateOpTokenAttrs func.func @CommunicateOpTokenAttrs() -> () { // CHECK: _xla_original_oc_node_name = [[NODE_NAME1:.*]], _xla_token_input_nodes = ["_xla_token_arg_node"] %0 = "tf.XlaRecvFromHost"() {key = "recv_key", shape = #tf_type.shape<>} : () -> tensor<i32> // CHECK: _xla_original_oc_node_name = [[NODE_NAME2:.*]], _xla_token_input_nodes = {{\[}}[[NODE_NAME1]]{{\]}}
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/encapsulate_subgraphs_pass_test.cc
{"shapes", absl::Span<const DataType>({})}, {"_outside_compilation_subgraph", "O1"}, {"_xla_token_input_nodes", absl::Span<const string>({"_xla_token_arg_node"})}, {"_xla_original_oc_node_name", "outside_compilation_O1_host_compute"}}, {"c"}}, }, {{"f_0_retval_retval", "F:o:0"}}); {
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/jit/extract_outside_compilation_pass_test.cc
"_xla_token_input_nodes", &token_input_nodes)); std::vector<string> expected_token_input_nodes_0({"_xla_token_arg_node"}); EXPECT_EQ(token_input_nodes, expected_token_input_nodes_0); token_input_nodes.clear(); std::vector<string> expected_token_input_nodes_1( {"_xla_token_arg_node", "outside_compilation_0_host_compute"}); TF_CHECK_OK(GetNodeAttr(AttrSlice(host_compute_1->attrs()),
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/mark_ops_for_outside_compilation.mlir
// CHECK-SAME:_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"] "tf.XlaHostCompute"(%cst) <{ancestors = [], cost_estimate_ns = 1000000 : i64, key = "_host_callback", recv_key = "", send_key = "", shapes = [], tpu_core = 0 : i64}> {_xla_original_oc_node_name = "hcb0", _xla_token_input_nodes = ["_xla_token_arg_node"]} : (tensor<i32>) -> () tf_device.return
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 16:22:32 UTC 2024 - 29.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
func.return %2336#0, %2336#1, %2336#2, %2336#3: tensor<1x1120x8xi32>, tensor<1x1120x8xi32>,...
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0)