Searched defs:grad_input (Results 1 – 2 of 2) sorted by relevance
28 auto grad_input = grad_outputs[i]; in Compute() local
352 mlir::OpOperand& grad_input = max_pool_grad_op->getOpOperand(2); in HandleMaxPoolGradOp() local