Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/
H A 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()
200 const absl::flat_hash_map<int, const Node*>& const_input_index_to_node) { in ReplaceRetvalInputWithArg() argument
215 for (const auto& iter : const_input_index_to_node) { in ReplaceRetvalInputWithArg()
230 const absl::flat_hash_map<int, const Node*>& const_input_index_to_node, in PropagateConstIntoFuncAttr() argument
249 ReplaceArgUsageWithConstNode(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
252 ReplaceRetvalInputWithArg(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
286 absl::flat_hash_map<int, const Node*> const_input_index_to_node; in PropagateConstIntoIfNode() local
291 const_input_index_to_node[i - 1] = input_node; in PropagateConstIntoIfNode()
294 if (const_input_index_to_node.empty()) { in PropagateConstIntoIfNode()
[all …]