Searched refs:input_to_forward (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | constant_folding.h | 110 void ReplaceOperationWithIdentity(int input_to_forward,
113 void ReplaceOperationWithSnapshot(int input_to_forward,
|
| D | dependency_optimizer.cc | 408 const string& input_to_forward = node->input(i); in OptimizeNode() local
409 CHECK(!IsControlInput(input_to_forward)); in OptimizeNode()
415 new_input = input_to_forward; in OptimizeNode()
421 new_input = AsControlDependency(NodeName(input_to_forward)); in OptimizeNode()
|
| D | constant_folding.cc | 1878 int input_to_forward, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithIdentity() argument
1880 if (input_to_forward < 0 || input_to_forward >= node->input_size()) return; in ReplaceOperationWithIdentity()
1888 node->mutable_input()->SwapElements(0, input_to_forward); in ReplaceOperationWithIdentity()
1903 int input_to_forward, const GraphProperties& properties, NodeDef* node, in ReplaceOperationWithSnapshot() argument
1908 ReplaceOperationWithIdentity(input_to_forward, properties, node, graph); in ReplaceOperationWithSnapshot()
1919 node->mutable_input()->SwapElements(0, input_to_forward); in ReplaceOperationWithSnapshot()
|