Searched refs:stateful_placements_ (Results 1 – 4 of 4) sorted by relevance
167 return stateful_placements_; in GetStatefulPlacements()180 std::unordered_map<string, string> stateful_placements_; // Immutable after variable
61 : stateful_placements_(options.stateful_placements), in GraphExecutionState()205 combined_options.stateful_placements = stateful_placements_; in Extend()232 stateful_placements_[n->name()] = n->assigned_device_name(); in SaveStatefulNodes()240 auto iter = stateful_placements_.find(n->name()); in RestoreStatefulNodes()241 if (iter != stateful_placements_.end()) { in RestoreStatefulNodes()
1475 prune_options.stateful_placements = stateful_placements_; in CreateGraphs()1513 auto iter = stateful_placements_.find(node_name); in CreateGraphs()1514 if (iter == stateful_placements_.end()) { in CreateGraphs()1515 stateful_placements_.insert(std::make_pair(node_name, placement)); in CreateGraphs()1524 stateful_placements_ = execution_state->GetStatefulPlacements(); in CreateGraphs()
373 std::unordered_map<string, string> stateful_placements_