Fix argument buffer reuse for MaxPoolingGrad[Grad]WithArgmaxOp
Both operations can destructively update their first input (index 0) but not their second (index 1). This is now reflected in the code, as well. PiperOrigin-RevId: 237205215
Loading
Please sign in to comment