| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| dependency_optimizer.h | 77 std::unique_ptr<NodeMap> node_map_; member in class:tensorflow::grappler::DependencyOptimizer
|
| constant_folding.h | 249 std::unique_ptr<NodeMap> node_map_; member in class:tensorflow::grappler::ConstantFolding
|
| scoped_allocator_optimizer.h | 66 NodeMap* node_map() { return node_map_.get(); }
105 std::unique_ptr<NodeMap> node_map_; member in class:tensorflow::grappler::ScopedAllocatorOptimizer
|
| arithmetic_optimizer.h | 132 std::unique_ptr<NodeMap> node_map_; member in class:tensorflow::grappler::ArithmeticOptimizer
|
| loop_optimizer.cc | 76 std::unique_ptr<NodeMap> node_map_; member in class:tensorflow::grappler::__anon45154::LoopInvariantNodeMotionOptimizer
89 auto consumers = node_map_->GetOutputs(node->name());
104 node_map_->AddOutput(NodeName(enter_input), consumer->name());
105 node_map_->RemoveOutput(node->name(), consumer->name());
110 node_map_->AddOutput(NodeName(control_input), consumer->name());
126 node_map_->RemoveInputs(node->name());
134 const_node = node_map_->GetNode(const_node_name);
141 node_map_->AddNode(const_node->name(), const_node);
143 auto consumers = node_map_->GetOutputs(node->name());
153 node_map_->AddOutput(const_node->name(), consumer->name())
[all...] |
| layout_optimizer.cc | 405 node_map_(node_map) {}
411 node_map_->AddNode(name, node);
437 node_map_->AddNode(name, node);
466 NodeMap* node_map_; member in class:tensorflow::grappler::__anon45152::GraphProcessor
555 auto outputs = node_map_->GetOutputs(node_->name());
613 auto input_node = node_map_->GetNode(node_->input(input_index));
624 node_map_->AddNode(node_name, added_node);
625 node_map_->AddOutput(node_name, node_->name());
646 auto input_node = node_map_->GetNode(node_->input(pos));
655 node_map_->UpdateOutput(NodeName(node_->input(pos)), node_->name()
[all...] |
| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| transform_utils.h | 179 std::map<string, const NodeDef*> node_map_; member in class:tensorflow::graph_transforms::GraphMatcher
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| virtual_scheduler.h | 298 return &node_map_;
331 std::unordered_map<const NodeDef*, NodeState> node_map_; member in class:tensorflow::grappler::VirtualScheduler
|
