Searched refs:RemoveOutput (Results 1 – 8 of 8) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.cc | 131 void NodeMap::RemoveOutput(const string& node_name, const string& output_name) { in RemoveOutput() function in tensorflow::grappler::NodeMap 137 RemoveOutput(NodeName(old_input_name), node_name); in UpdateInput() 144 RemoveOutput(NodeName(input), node->name()); in RemoveInputs()
|
D | utils.h | 59 void RemoveOutput(const string& node_name, const string& output_name);
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | dependency_optimizer.cc | 48 node_map->RemoveOutput(NodeName(input), node->name()); in RemoveInput() 255 node_map_->RemoveOutput(node_name, fanout->name()); in OptimizeNode() 570 node_map_->RemoveOutput(source_node.name(), target_node->name()); in TransitiveReduction()
|
D | loop_optimizer.cc | 105 node_map_->RemoveOutput(node->name(), consumer->name()); in HandleInvariantEnter() 154 node_map_->RemoveOutput(node->name(), consumer->name()); in HandleConst()
|
D | scoped_allocator_optimizer.cc | 182 node_map->RemoveOutput(from_node_name, to_node->name()); in RemoveEdge() 517 node_map->RemoveOutput(old_op->name(), n->name()); in RewireSubgraph()
|
D | constant_folding.cc | 140 node_map->RemoveOutput(NodeName(old_input), node->name()); in MaybeRemoveControlInput() 498 node_map_->RemoveOutput(node->name(), output->name()); in MaterializeShapes() 3286 node_map_->RemoveOutput(node->input(j), node->name()); in MergeConcat() 3345 node_map_->RemoveOutput(node->name(), output->name()); in AddQuantizedMatMulMinMaxOutConstNodes()
|
D | arithmetic_optimizer.cc | 1070 ctx().node_map->RemoveOutput(NodeName(old_input_0), node->name()); in UpdateInputs() 1071 ctx().node_map->RemoveOutput(NodeName(old_input_1), node->name()); in UpdateInputs()
|
D | layout_optimizer.cc | 707 node_map_->RemoveOutput(node_->name(), output->name()); in AddTransformToOutputs()
|