Searched refs:input_to_swap (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 712 Status BuildSwapPair(NodeDef* node, int input_to_swap, in BuildSwapPair() argument
719 return errors::InvalidArgument("Can't swap input ", input_to_swap, in BuildSwapPair()
727 InputTypeForNode(*node, *op_def, input_to_swap, &input_type)); in BuildSwapPair()
729 return errors::InvalidArgument("Can't swap input ", input_to_swap, in BuildSwapPair()
734 string tensor_to_swap = strings::StrCat(node->name(), "_", input_to_swap); in BuildSwapPair()
739 return errors::InvalidArgument("Input ", input_to_swap, " of node ", in BuildSwapPair()
|