| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 266 const auto& tmp = node_map_->GetOutputs(node->name()); in ForwardInputs()
274 const NodeDef* input_node = node_map_->GetNode(NodeName(input)); in ForwardInputs()
301 node_map_->AddOutput(NodeName(input), consumer->name()); in ForwardInputs()
362 return node_map_->NodeExists(OptimizedNodeName(node, suffix)); in OptimizedNodeExists()
382 const NodeDef* node = node_map_->GetNode(node_name_or_input); in GetTensorFromConstNode()
448 AddControlDependency(node->input(0), graph_, node_map_.get()); in MaterializeShapes()
449 node_map_->UpdateInput(node->name(), node->input(0), ctrl_dep); in MaterializeShapes()
456 const NodeDef* array = CHECK_NOTNULL(node_map_->GetNode(node->input(0))); in MaterializeShapes()
463 CHECK_NOTNULL(node_map_->GetNode(array->input(0))); in MaterializeShapes()
487 graph_, node_map_.get())); in MaterializeShapes()
[all …]
|
| D | dependency_optimizer.cc | 71 const NodeDef* input = node_map_->GetNode(NodeName(node.input(0))); in SafeToRemoveIdentity()
79 for (const auto& consumer : node_map_->GetOutputs(node.name())) { in SafeToRemoveIdentity()
95 if (HasRegularOutputs(node, *node_map_)) { in SafeToConvertToNoOp()
258 const auto output_nodes = node_map_->GetOutputs(node_name); in OptimizeNode()
277 node_map_->RemoveOutput(node_name, fanout->name()); in OptimizeNode()
281 if (node_map_->GetOutputs(node_name).empty() && fetch_nodes_known_ && in OptimizeNode()
311 old_input, optimized_graph_, node_map_.get()); in OptimizeNode()
314 node_map_->UpdateInput(node_name, old_input, ctrl_input); in OptimizeNode()
315 const NodeDef* old_input_node = node_map_->GetNode(old_input); in OptimizeNode()
379 NodeDef* input_node = node_map_->GetNode(node->input(i)); in OptimizeNode()
[all …]
|
| D | loop_optimizer.cc | 76 std::unique_ptr<NodeMap> node_map_; member in tensorflow::grappler::__anon3d4744950111::LoopInvariantNodeMotionOptimizer
88 auto consumers = node_map_->GetOutputs(node->name()); in HandleInvariantEnter()
103 node_map_->AddOutput(NodeName(enter_input), consumer->name()); in HandleInvariantEnter()
104 node_map_->RemoveOutput(node->name(), consumer->name()); in HandleInvariantEnter()
109 node_map_->AddOutput(NodeName(control_input), consumer->name()); in HandleInvariantEnter()
125 node_map_->RemoveInputs(node->name()); in HandleConst()
133 const_node = node_map_->GetNode(const_node_name); in HandleConst()
140 node_map_->AddNode(const_node->name(), const_node); in HandleConst()
142 auto consumers = node_map_->GetOutputs(node->name()); in HandleConst()
152 node_map_->AddOutput(const_node->name(), consumer->name()); in HandleConst()
[all …]
|
| D | layout_optimizer.cc | 413 node_map_(node_map) {} in GraphProcessor()
419 node_map_->AddNode(name, node); in AddNodePermConst()
445 node_map_->AddNode(name, node); in AddNodeConstScalar()
474 NodeMap* node_map_; member in tensorflow::grappler::__anon5a8ea6d90111::GraphProcessor
567 auto outputs = node_map_->GetOutputs(node_->name()); in HasOutputs()
625 auto input_node = node_map_->GetNode(node_->input(input_index)); in UpdateAttrValueOfInput()
636 node_map_->AddNode(node_name, added_node); in UpdateAttrValueOfInput()
637 node_map_->AddOutput(node_name, node_->name()); in UpdateAttrValueOfInput()
658 auto input_node = node_map_->GetNode(node_->input(pos)); in AddLayoutTransposeToInputs()
667 node_map_->UpdateOutput(NodeName(node_->input(pos)), node_->name(), in AddLayoutTransposeToInputs()
[all …]
|
| D | scoped_allocator_optimizer.h | 76 NodeMap* node_map() { return node_map_.get(); } in node_map()
120 std::unique_ptr<NodeMap> node_map_; variable
|
| D | dependency_optimizer.h | 80 std::unique_ptr<NodeMap> node_map_; variable
|
| D | arithmetic_optimizer.h | 136 std::unique_ptr<NodeMap> node_map_; variable
|
| D | constant_folding.h | 335 std::unique_ptr<NodeMap> node_map_; variable
|
| D | arithmetic_optimizer.cc | 3672 graph_properties_.get(), node_map_.get(), in SimplifyArithmeticOps()
3755 NodeDef* simplified_node = node_map_->GetNode(simplified_tensor); in SimplifyArithmeticOps()
3764 node_map_->GetOutputsOrderedByNodeName(node->name()); in SimplifyArithmeticOps()
3778 node_map_->UpdateInput(consumer->name(), node->name(), in SimplifyArithmeticOps()
3796 node_map_.reset(new NodeMap(optimized_graph_)); in Optimize()
|
| D | scoped_allocator_optimizer.cc | 925 node_map_ = absl::make_unique<NodeMap>(optimized_graph); in Optimize()
|
| /external/rust/crates/grpcio-sys/grpc/src/core/lib/channel/ |
| D | channelz_registry.cc | 61 node_map_[node->uuid_] = node; in InternalRegister()
68 node_map_.erase(uuid); in InternalUnregister()
76 auto it = node_map_.find(uuid); in InternalGet()
77 if (it == node_map_.end()) return nullptr; in InternalGet()
90 for (auto it = node_map_.lower_bound(start_channel_id); in InternalGetTopChannels()
91 it != node_map_.end(); ++it) { in InternalGetTopChannels()
129 for (auto it = node_map_.lower_bound(start_server_id); in InternalGetServers()
130 it != node_map_.end(); ++it) { in InternalGetServers()
167 for (auto& p : node_map_) { in InternalLogAllEntities()
|
| D | channelz_registry.h | 92 std::map<intptr_t, BaseNode*> node_map_; variable
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | virtual_scheduler.cc | 169 node_map_ = node_map; in Init()
235 return FirstReadyCmp(node_map_, a, b); in Greater()
247 return FirstReadyCmp(node_map_, a, b); in Greater()
273 node_map_ = node_map; in Init()
286 const auto& device = node_map_->at(node).device_name; in AddNode()
303 candidates.emplace_back(op, node_map_->at(op).time_ready); in GetCurrNode()
308 candidates.emplace_back(send, node_map_->at(send).time_ready); in GetCurrNode()
312 candidates.emplace_back(recv, node_map_->at(recv).time_ready); in GetCurrNode()
351 const auto device = node_map_->at(node).device_name; in RemoveCurrNode()
406 node_map_.clear(); in Init()
[all …]
|
| D | virtual_scheduler.h | 240 const std::unordered_map<const NodeDef*, NodeState>* node_map_; variable
308 const std::unordered_map<const NodeDef*, NodeState>* node_map_; variable
369 return &node_map_; in GetNodeStates()
386 return node_map_; in GetNodeMap()
419 std::unordered_map<const NodeDef*, NodeState> node_map_; variable
|
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| D | transform_utils.h | 179 std::map<string, const NodeDef*> node_map_; variable
|
| D | transform_utils.cc | 288 MapNamesToNodes(graph_def_, &node_map_); in GraphMatcher()
351 const NodeDef& input_node = *(node_map_[input_node_name]); in DoesOpTypeMatch()
|