Commit 7b79e43b authored by A. Unique TensorFlower's avatar A. Unique TensorFlower Committed by TensorFlower Gardener
Browse files

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
parent 83903c9d
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment