Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dconst_analysis.cc186 std::vector<bool>* compile_time_const_nodes, in BackwardsConstAnalysis() argument
189 if (!compile_time_const_nodes && g.GetConstArgIndicesCache().has_value() && in BackwardsConstAnalysis()
200 if (compile_time_const_nodes) { in BackwardsConstAnalysis()
201 CHECK_EQ(compile_time_const_nodes->size(), g.num_node_ids()); in BackwardsConstAnalysis()
204 compile_time_const_nodes = &compile_time_const_nodes_impl; in BackwardsConstAnalysis()
218 if ((*compile_time_const_nodes)[node->id()]) { in BackwardsConstAnalysis()
240 (*compile_time_const_nodes)[pred->src()->id()] = true; in BackwardsConstAnalysis()
270 (*compile_time_const_nodes)[edge->src()->id()] = true; in BackwardsConstAnalysis()
Dconst_analysis.h40 std::vector<bool>* compile_time_const_nodes,
/external/tensorflow/tensorflow/compiler/jit/
Dpartially_decluster_pass.cc296 std::vector<bool> compile_time_const_nodes(graph->num_node_ids()); in PartiallyDeclusterGraph() local
303 &compile_time_const_nodes, in PartiallyDeclusterGraph()
310 if (!compile_time_const_nodes[n->id()]) { in PartiallyDeclusterGraph()
Dmark_for_compilation_pass.cc1135 std::vector<bool> compile_time_const_nodes(graph_->num_node_ids(), false); in FindCompilationCandidates() local
1138 &compile_time_const_nodes, lib_runtime)); in FindCompilationCandidates()
1226 if (compile_time_const_nodes[node->id()]) { in FindCompilationCandidates()