Searched refs:input_to_forward (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 107 void ReplaceOperationWithIdentity(int input_to_forward,
110 void ReplaceOperationWithSnapshot(int input_to_forward,
|
| D | dependency_optimizer.cc | 416 const string& input_to_forward = node->input(i); in OptimizeNode() local
417 CHECK(!IsControlInput(input_to_forward)); in OptimizeNode()
423 new_input = input_to_forward; in OptimizeNode()
429 new_input = AsControlDependency(NodeName(input_to_forward)); in OptimizeNode()
|
| D | constant_folding.cc | 1927 int input_to_forward, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithIdentity() argument
1929 if (input_to_forward < 0 || input_to_forward >= node->input_size()) return; in ReplaceOperationWithIdentity()
1937 node->mutable_input()->SwapElements(0, input_to_forward); in ReplaceOperationWithIdentity()
1952 int input_to_forward, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithSnapshot() argument
1957 ReplaceOperationWithIdentity(input_to_forward, properties, node, graph); in ReplaceOperationWithSnapshot()
1968 node->mutable_input()->SwapElements(0, input_to_forward); in ReplaceOperationWithSnapshot()
|