Searched refs:input_to_swap (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | memory_optimizer.cc | 743 Status BuildSwapPair(NodeDef* node, int input_to_swap, in BuildSwapPair() argument
750 return errors::InvalidArgument("Can't swap input ", input_to_swap, in BuildSwapPair()
758 InputTypeForNode(*node, *op_def, input_to_swap, &input_type)); in BuildSwapPair()
760 return errors::InvalidArgument("Can't swap input ", input_to_swap, in BuildSwapPair()
765 string tensor_to_swap = strings::StrCat(node->name(), "_", input_to_swap); in BuildSwapPair()
770 return errors::InvalidArgument("Input ", input_to_swap, " of node ", in BuildSwapPair()
|
| D | constant_folding.cc | 3254 int input_to_swap = -1; in ConstantPushDownBiasAdd() local
3259 input_to_swap = matrix_idx; in ConstantPushDownBiasAdd()
3262 input_to_swap = vector_idx; in ConstantPushDownBiasAdd()
3264 if (input_to_swap == -1) return false; in ConstantPushDownBiasAdd()
3266 node_map_->GetNode(ctx.op_child->input(input_to_swap)); in ConstantPushDownBiasAdd()
3270 ctx.op_child->input(input_to_swap)); in ConstantPushDownBiasAdd()
3272 if (ctx.op_child->input(input_to_swap) != in ConstantPushDownBiasAdd()
3273 ctx.op_child->input(1 - input_to_swap)) { in ConstantPushDownBiasAdd()
3274 node_map_->RemoveOutput(ctx.op_child->input(input_to_swap), in ConstantPushDownBiasAdd()
3278 *ctx.op_child->mutable_input(input_to_swap)); in ConstantPushDownBiasAdd()
|