Searched refs:UpdateOutput (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/grappler/ |
D | utils.h | 61 void UpdateOutput(const string& node_name, const string& old_output_name,
|
D | utils.cc | 152 void NodeMap::UpdateOutput(const string& node_name, in UpdateOutput() function in tensorflow::grappler::NodeMap
|
/external/tensorflow/tensorflow/core/grappler/optimizers/ |
D | dependency_optimizer.cc | 658 node_map_->UpdateOutput(input_name, node->name(), in GroupCrossDeviceControlEdges()
|
D | layout_optimizer.cc | 655 node_map_->UpdateOutput(NodeName(node_->input(pos)), node_->name(), in AddLayoutTransposeToInputs() 1000 node_map_->UpdateOutput(NodeName(added_node->input(0)), node_->name(), in AddDataFormatTranformToParamInput() 1416 node_map_->UpdateOutput(NodeName(node_->input(vector_index)), in CustomizedProcessing()
|
D | constant_folding.cc | 2799 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul() 3120 node_map_->UpdateOutput(NodeName(node->input(i)), node->name(), in PartialAssocOpConstFolding() 3225 node_map_->UpdateOutput(NodeName(node->input(i)), node->name(), in PartialConcatConstFolding()
|
D | arithmetic_optimizer.cc | 1685 ctx().node_map->UpdateOutput(NodeName(orig_input), split_name, in HoistChainForSplit()
|