HomeSort by relevance Sort by last modified time
    Searched full:recomputed_node_components (Results 1 - 1 of 1) sorted by null

  /external/tensorflow/tensorflow/core/grappler/optimizers/
memory_optimizer.cc 209 std::unordered_map<const NodeDef*, int> recomputed_node_components; local
223 recomputed_node_components[original_recompute_node] =
242 recomputed_node_components.find(original_recompute_node);
243 if (recomputed_component_iterator != recomputed_node_components.end()) {
253 auto child_component_iterator = recomputed_node_components.find(output);
254 CHECK(child_component_iterator != recomputed_node_components.end());
261 recomputed_node_components[original_recompute_node] = max_component;
263 return recomputed_node_components;
375 std::unordered_map<const NodeDef*, int> recomputed_node_components = local
385 recomputed_node_components, graph)
    [all...]

Completed in 77 milliseconds