Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_util.cc171 const absl::flat_hash_map<int, const Node*>& const_input_index_to_node) { in ReplaceArgUsageWithConstNode() argument
182 for (const auto& iter : const_input_index_to_node) { in ReplaceArgUsageWithConstNode()
202 const absl::flat_hash_map<int, const Node*>& const_input_index_to_node) { in ReplaceRetvalInputWithArg() argument
217 for (const auto& iter : const_input_index_to_node) { in ReplaceRetvalInputWithArg()
232 const absl::flat_hash_map<int, const Node*>& const_input_index_to_node, in PropagateConstIntoFuncAttr() argument
251 ReplaceArgUsageWithConstNode(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
254 ReplaceRetvalInputWithArg(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
288 absl::flat_hash_map<int, const Node*> const_input_index_to_node; in PropagateConstIntoIfNode() local
293 const_input_index_to_node[i - 1] = input_node; in PropagateConstIntoIfNode()
296 if (const_input_index_to_node.empty()) { in PropagateConstIntoIfNode()
[all …]