Searched refs:max_pool_output (Results 1 – 2 of 2) sorted by relevance
603 max_pool_output = max_pool(conv1_output)604 conv2_output = conv2(max_pool_output)613 max_pool_output.size(),
348 mlir::OpOperand& max_pool_output = max_pool_grad_op->getOpOperand(1); in HandleMaxPoolGradOp() local350 ExtractRequiredLayoutFromOperand(max_pool_output.get())); in HandleMaxPoolGradOp()