- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for oc_if_if (0.12 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass_test.cc
// Verify we have an "If" to choose outside compilation between then_branch // and else_branch, and it has `recv_if_pred_node` as cond input. Node *if_oc_node = node_name_index["oc_if_if"]; EXPECT_NE(if_oc_node, nullptr); Node *if_oc_node_cond_input; TF_CHECK_OK(if_oc_node->input_node(0, &if_oc_node_cond_input)); EXPECT_EQ(if_oc_node_cond_input, recv_if_pred_node);
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/jit/extract_outside_compilation_pass.cc
const string& then_branch_host_func_name, const string& else_branch_host_func_name) { Graph host_graph(fld); string outside_compilation_name = absl::StrCat("oc_if_", if_node_name); AttrValue device_ordinal_value; device_ordinal_value.set_placeholder("_device_ordinal"); // Step 1: add key placeholder node. TF_ASSIGN_OR_RETURN( Node * key_placeholder,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)