Searched refs:UpdateInput (Results 1 – 8 of 8) sorted by relevance
72 void UpdateInput(int i, tensorflow::TensorHandle* h);163 inline void EagerOperation::UpdateInput(int i, tensorflow::TensorHandle* h) { in UpdateInput() function
187 op->UpdateInput(i, handle); in ValidateInputTypeAndPlacement()392 op->UpdateInput(i, handle); in EagerLocalExecute()687 op->UpdateInput(i, handle); in EagerRemoteExecute()
692 node_map_->UpdateInput(output->name(), node_name, out[port]->name()); in MaterializeBroadcastGradientArgs()798 node_map_->UpdateInput(node->name(), indices->name(), in MaterializeReductionIndices()853 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in MaterializeConstantValuedNode()1517 node_map_->UpdateInput(node_name, NodeName(output->input(i)), in FoldNode()1524 node_map_->UpdateInput(output->name(), NodeName(output->input(i)), in FoldNode()1774 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithIdentity()1805 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceOperationWithSnapshot()1855 node_map_->UpdateInput(node->name(), node->input(i), ctrl_dep); in ReplaceBinaryOperationWithBroadcastTo()1871 node_map_->UpdateInput(node->name(), node->input(1), ctrl_dep); in ReplaceDivisionOfOnesByReciprocal()1882 node_map_->UpdateInput(node->name(), node->input(1), ctrl_dep); in ReplaceSubtractionFromZeroByNegation()[all …]
1154 ctx().node_map->UpdateInput(tail->name(), first_transpose->name(), in TrySimplify()1241 ctx().node_map->UpdateInput(tail->name(), involution->name(), in TrySimplify()1292 ctx().node_map->UpdateInput(bitcast->name(), bitcast->input(0), in TrySimplify()1348 ctx().node_map->UpdateInput(node->name(), node->input(1), y->input(0)); in TrySimplify()1355 ctx().node_map->UpdateInput(node->name(), node->input(0), x->input(0)); in TrySimplify()1694 ctx().node_map->UpdateInput(concat_name, concat_input, tail_input); in HoistChainForConcat()1702 ctx().node_map->UpdateInput(tail->name(), tail_input, concat_name); in HoistChainForConcat()1726 ctx().node_map->UpdateInput(split_node->name(), orig_input, in HoistChainForSplit()1768 ctx().node_map->UpdateInput(consumer->name(), node_name, new_input); in UpdateConsumers()1929 ctx().node_map->UpdateInput(node->name(), input->name(), input->input(0)); in TrySimplify()[all …]
292 node_map_->UpdateInput(node_name, old_input, ctrl_input); in OptimizeNode()394 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()400 node_map_->UpdateInput(consumer->name(), old_input.ToString(), in OptimizeNode()
239 node_map->UpdateInput(op->name(), input->name(), identity_name); in MaybeRewriteInput()744 node_map->UpdateInput(n->name(), old_op->name(), sas_name); in RewireSubgraph()
57 void UpdateInput(const string& node_name, const string& old_input_name,
138 void NodeMap::UpdateInput(const string& node_name, const string& old_input_name, in UpdateInput() function in tensorflow::grappler::NodeMap