- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for node_name_index (0.42 sec)
-
tensorflow/compiler/jit/rearrange_function_argument_pass_test.cc
ASSERT_EQ(f1_rewritten->signature().output_arg_size(), 1); EXPECT_EQ(f1_rewritten->signature().output_arg(0).type(), DT_BOOL); // Check node "if" input and output edges. auto node_name_index = g->BuildNodeNameIndex(); const Node *if_node = node_name_index.at("if"); ASSERT_NE(if_node, nullptr); const Node *input_node; TF_CHECK_OK(if_node->input_node(1, &input_node)); EXPECT_EQ(input_node->name(), "arg1");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Feb 09 11:36:41 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
TF_RETURN_IF_ERROR(GetNodeAttr( n->attrs(), kOutsideCompilationSrcOutputAttrName, &node_src_output)); auto iter = node_name_index.find(node_name); if (iter == node_name_index.end()) { return errors::Internal( "Cannot find original node for oc -> host placeholder node ", node_name); } // Change all usage node to use the original node instead. Node* original_node = iter->second;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
bool* modified) { if (VLOG_IS_ON(4)) { DumpGraphToFile("extract_outside_compilation_before", *g, fld); } *modified = false; auto node_name_index = g->BuildNodeNameIndex(); for (auto& iter : clusters) { string xla_cluster_name = iter.first; Node* n = iter.second.node; auto const& func_name_attrs = iter.second.func_name_attrs;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0)