Commit 1db31998 authored by Sanjoy Das's avatar Sanjoy Das Committed by TensorFlower Gardener
Browse files

[TF:XLA] Don't ignore data_format in XLA's {Avg|Max}Pool3d{Grad} lowering 

I didn't add a dedicated test for this because soon we'll be exercising this via
python/kernel_tests/pooling_ops_3d_test.py (that's how I found this problem).

PiperOrigin-RevId: 223995407
parent 551248d4
Show whitespace changes
Inline Side-by-side
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment