Home
last modified time | relevance | path

Searched refs:const_nodes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis_test.cc50 std::vector<bool> const_nodes(root.graph()->num_node_ids(), false); in TEST() local
51 TF_ASSERT_OK(BackwardsConstAnalysis(*root.graph(), &const_args, &const_nodes, in TEST()
60 EXPECT_FALSE(const_nodes[arg0.node()->id()]); in TEST()
61 EXPECT_TRUE(const_nodes[arg1.node()->id()]); in TEST()
62 EXPECT_FALSE(const_nodes[arg2.node()->id()]); in TEST()
63 EXPECT_TRUE(const_nodes[arg3.node()->id()]); in TEST()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dpin_to_host_optimizer.cc326 std::vector<std::pair<NodeDef*, string>> const_nodes; in Optimize() local
342 const_nodes.emplace_back(&node, node.device()); in Optimize()
350 for (auto& it : const_nodes) { in Optimize()
Dconstant_folding.cc1487 std::vector<NodeDef> const_nodes; in FoldNode() local
1489 EvaluateOneFoldable(*node, &const_nodes, result_too_large)); in FoldNode()
1493 for (int i = 0, end = const_nodes.size(); i < end; i++) { in FoldNode()
1494 NodeDef* const_node = &const_nodes[i]; in FoldNode()
1532 if (const_nodes.size() == 1) { in FoldNode()
1565 if (const_nodes.size() > 1) { in FoldNode()
1581 } else if (port < static_cast<int>(const_nodes.size()) && in FoldNode()
1582 !const_nodes[port].name().empty()) { in FoldNode()
1585 const_nodes[port].name()); in FoldNode()
1586 *output->mutable_input(i) = const_nodes[port].name(); in FoldNode()
Dfunction_optimizer.cc347 absl::flat_hash_map<string, const NodeDef*> const_nodes; in InferTrulyConstNodes() local
350 const_nodes[node.name()] = &node; in InferTrulyConstNodes()
354 return const_nodes; in InferTrulyConstNodes()