Searched refs:UpdateInput (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.cc | 662 node_map_->UpdateInput(output->name(), node_name, out[port]->name()); in MaterializeBroadcastGradientArgs()
768 node_map_->UpdateInput(node->name(), indices->name(), in MaterializeReductionIndices()
823 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in MaterializeConstantValuedNode()
1405 node_map_->UpdateInput(node_name, NodeName(output->input(i)), in FoldNode()
1412 node_map_->UpdateInput(output->name(), NodeName(output->input(i)), in FoldNode()
1659 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithIdentity()
1690 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithSnapshot()
1702 node_map_->UpdateInput(node->name(), node->input(1), ctrl_dep); in ReplaceDivisionOfOnesByReciprocal()
1713 node_map_->UpdateInput(node->name(), node->input(1), ctrl_dep); in ReplaceSubtractionFromZeroByNegation()
1742 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithConstant()
[all …]
|
| D | arithmetic_optimizer.cc | 1129 ctx().node_map->UpdateInput(tail->name(), first_transpose->name(), in TrySimplify()
1216 ctx().node_map->UpdateInput(tail->name(), involution->name(), in TrySimplify()
1267 ctx().node_map->UpdateInput(bitcast->name(), bitcast->input(0), in TrySimplify()
1323 ctx().node_map->UpdateInput(node->name(), node->input(1), y->input(0)); in TrySimplify()
1330 ctx().node_map->UpdateInput(node->name(), node->input(0), x->input(0)); in TrySimplify()
1637 ctx().node_map->UpdateInput(concat_name, concat_input, tail_input); in HoistChainForConcat()
1645 ctx().node_map->UpdateInput(tail->name(), tail_input, concat_name); in HoistChainForConcat()
1669 ctx().node_map->UpdateInput(split_node->name(), orig_input, in HoistChainForSplit()
1710 ctx().node_map->UpdateInput(consumer->name(), node_name, new_input); in UpdateConsumers()
1863 ctx().node_map->UpdateInput(node->name(), input->name(), input->input(0)); in TrySimplify()
[all …]
|
| D | dependency_optimizer.cc | 292 node_map_->UpdateInput(node_name, old_input, ctrl_input); in OptimizeNode()
392 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()
398 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | utils.h | 55 void UpdateInput(const string& node_name, const string& old_input_name,
|
| D | utils.cc | 135 void NodeMap::UpdateInput(const string& node_name, const string& old_input_name, in UpdateInput() function in tensorflow::grappler::NodeMap
|