Home
last modified time | relevance | path

Searched refs:UpdateOutput (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/
Dutils.h61 void UpdateOutput(const string& node_name, const string& old_output_name,
Dutils.cc152 void NodeMap::UpdateOutput(const string& node_name, in UpdateOutput() function in tensorflow::grappler::NodeMap
/external/tensorflow/tensorflow/core/grappler/optimizers/
Ddependency_optimizer.cc658 node_map_->UpdateOutput(input_name, node->name(), in GroupCrossDeviceControlEdges()
Dlayout_optimizer.cc655 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()
Dconstant_folding.cc2799 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()
Darithmetic_optimizer.cc1685 ctx().node_map->UpdateOutput(NodeName(orig_input), split_name, in HoistChainForSplit()