Searched refs:recomputed_node_components (Results 1 – 1 of 1) sorted by relevance
217 std::unordered_map<const NodeDef*, int> recomputed_node_components; in GetMaxDownstreamComponents() local231 recomputed_node_components[original_recompute_node] = in GetMaxDownstreamComponents()250 recomputed_node_components.find(original_recompute_node); in GetMaxDownstreamComponents()251 if (recomputed_component_iterator != recomputed_node_components.end()) { in GetMaxDownstreamComponents()261 auto child_component_iterator = recomputed_node_components.find(output); in GetMaxDownstreamComponents()262 CHECK(child_component_iterator != recomputed_node_components.end()); in GetMaxDownstreamComponents()269 recomputed_node_components[original_recompute_node] = max_component; in GetMaxDownstreamComponents()271 return recomputed_node_components; in GetMaxDownstreamComponents()383 std::unordered_map<const NodeDef*, int> recomputed_node_components = in RecomputeSubgraph() local393 recomputed_node_components, graph); in RecomputeSubgraph()