Searched refs:const_input_index_to_node (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/compiler/tf2xla/ |
| D | tf2xla_util.cc | 113 const std::unordered_map<int, const Node*>& const_input_index_to_node) { in ReplaceArgUsageWithConstNode() argument
124 for (const auto& iter : const_input_index_to_node) { in ReplaceArgUsageWithConstNode()
176 const std::unordered_map<int, const Node*>& const_input_index_to_node, in PropagateConstIntoFuncAttr() argument
199 ReplaceArgUsageWithConstNode(func_graph, const_input_index_to_node)); in PropagateConstIntoFuncAttr()
227 std::unordered_map<int, const Node*> const_input_index_to_node; in PropagateConstIntoIfNode() local
232 const_input_index_to_node[i - 1] = input_node; in PropagateConstIntoIfNode()
235 if (const_input_index_to_node.empty()) { in PropagateConstIntoIfNode()
244 if_node, attr_name, const_input_index_to_node, lookup_fld, fld)); in PropagateConstIntoIfNode()
258 std::unordered_map<int, const Node*> const_input_index_to_node; in PropagateConstIntoWhileNode() local
293 const_input_index_to_node[i] = input_node; in PropagateConstIntoWhileNode()
[all …]
|