Searched refs:UpdateOutput (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 215 void UpdateOutput(const string& node_name, const string& old_output_name, in UpdateOutput() function
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | layout_optimizer.cc | 667 node_map_->UpdateOutput(NodeName(node_->input(pos)), node_->name(), in AddLayoutTransposeToInputs()
1051 node_map_->UpdateOutput(NodeName(added_node->input(0)), node_->name(), in AddDataFormatTransformToParamInput()
1475 node_map_->UpdateOutput(NodeName(node_->input(vector_index)), in CustomizedProcessing()
|
| D | dependency_optimizer.cc | 724 node_map_->UpdateOutput(input_name, node->name(), in GroupCrossDeviceControlEdges()
|
| D | arithmetic_optimizer.cc | 1764 ctx().node_map->UpdateOutput(NodeName(orig_input), split_name, in HoistChainForSplit()
2323 ctx().node_map->UpdateOutput(a->name(), node->name(), new_op->name()); in TrySimplify()
2332 ctx().node_map->UpdateOutput(b->name(), node->name(), new_op->name()); in TrySimplify()
|
| D | constant_folding.cc | 3085 node_map_->UpdateOutput(node->name(), const_input, reciprocal_node->name()); in ReduceDivToReciprocalMul()
3635 node_map_->UpdateOutput(NodeName(node->input(i)), node->name(), in PartialAssocOpConstFolding()
|