Home
last modified time | relevance | path

Searched refs:UpdateInput (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dconstant_folding.cc449 node_map_->UpdateInput(node->name(), node->input(0), ctrl_dep); in MaterializeShapes()
717 node_map_->UpdateInput(output->name(), node_name, out[port]->name()); in MaterializeBroadcastGradientArgs()
823 node_map_->UpdateInput(node->name(), indices->name(), in MaterializeReductionIndices()
881 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in MaterializeConstantValuedNode()
1577 node_map_->UpdateInput(node_name, NodeName(output->input(i)), in FoldNode()
1584 node_map_->UpdateInput(output->name(), NodeName(output->input(i)), in FoldNode()
1866 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithBroadcastTo()
1896 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithIdentity()
1927 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithSnapshot()
1951 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithNoOp()
[all …]
Darithmetic_optimizer.cc1157 ctx().node_map->UpdateInput(tail->name(), first_transpose->name(), in TrySimplify()
1256 ctx().node_map->UpdateInput(tail->name(), involution->name(), in TrySimplify()
1307 ctx().node_map->UpdateInput(bitcast->name(), bitcast->input(0), in TrySimplify()
1365 ctx().node_map->UpdateInput(node->name(), node->input(1), y->input(0)); in TrySimplify()
1372 ctx().node_map->UpdateInput(node->name(), node->input(0), x->input(0)); in TrySimplify()
1716 ctx().node_map->UpdateInput(concat_name, concat_input, tail_input); in HoistChainForConcat()
1724 ctx().node_map->UpdateInput(tail->name(), tail_input, concat_name); in HoistChainForConcat()
1748 ctx().node_map->UpdateInput(split_node->name(), orig_input, in HoistChainForSplit()
1790 ctx().node_map->UpdateInput(consumer->name(), node_name, new_input); in UpdateConsumers()
1964 ctx().node_map->UpdateInput(node->name(), input->name(), input->input(0)); in TrySimplify()
[all …]
Ddependency_optimizer.cc314 node_map_->UpdateInput(node_name, old_input, ctrl_input); in OptimizeNode()
416 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()
422 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()
Dcommon_subgraph_elimination.cc250 node_map.UpdateInput(fanout->name(), node->name(), rep->name()); in DedupComputations()
Dscoped_allocator_optimizer.cc245 node_map->UpdateInput(op->name(), input->name(), identity_name); in MaybeRewriteInput()
771 node_map->UpdateInput(n->name(), old_op->name(), sas_name); in RewireSubgraph()
/external/tensorflow/tensorflow/core/common_runtime/eager/
Deager_operation.h167 void UpdateInput(int i, TensorHandle* h);
264 inline void EagerOperation::UpdateInput(int i, TensorHandle* h) { in UpdateInput() function
Dexecute.cc210 op->UpdateInput(i, h); in ValidateInputTypeAndPlacement()
224 op->UpdateInput(i, handle); in ValidateInputTypeAndPlacement()
779 op->UpdateInput(i, ret); in MaybePackInputTensor()
881 op->UpdateInput(i, handle); in EagerRemoteExecute()
/external/tensorflow/tensorflow/core/grappler/
Dutils.h200 void UpdateInput(const string& node_name, const string& old_input_name, in UpdateInput() function