Searched refs:const_nodes (Results 1 – 4 of 4) sorted by relevance
46 std::vector<bool> const_nodes(root.graph()->num_node_ids(), false); in TEST() local47 TF_ASSERT_OK(BackwardsConstAnalysis(*root.graph(), &const_args, &const_nodes, in TEST()56 EXPECT_FALSE(const_nodes[arg0.node()->id()]); in TEST()57 EXPECT_TRUE(const_nodes[arg1.node()->id()]); in TEST()58 EXPECT_FALSE(const_nodes[arg2.node()->id()]); in TEST()59 EXPECT_TRUE(const_nodes[arg3.node()->id()]); in TEST()
331 std::vector<std::pair<NodeDef*, string>> const_nodes; in Optimize() local347 const_nodes.emplace_back(&node, node.device()); in Optimize()354 for (auto& it : const_nodes) { in Optimize()
1324 std::vector<NodeDef> const_nodes; in FoldNode() local1326 EvaluateOneFoldable(*node, &const_nodes, result_too_large)); in FoldNode()1330 for (int i = 0; i < const_nodes.size(); i++) { in FoldNode()1331 NodeDef* const_node = &const_nodes[i]; in FoldNode()1361 if (const_nodes.size() == 1) { in FoldNode()1394 if (const_nodes.size() > 1) { in FoldNode()1409 } else if (port < const_nodes.size() && in FoldNode()1410 !const_nodes[port].name().empty()) { in FoldNode()1413 const_nodes[port].name()); in FoldNode()1414 *output->mutable_input(i) = const_nodes[port].name(); in FoldNode()
399 absl::flat_hash_map<string, const NodeDef*> const_nodes; in InferTrulyConstNodes() local402 const_nodes[node.name()] = &node; in InferTrulyConstNodes()406 return const_nodes; in InferTrulyConstNodes()