tree e82d7638396e160e46675ca2cfbd8837fa4d3e82
parent ea5e8c35bd2d5dab46bbca83de04f20a72508141
author Andy Ly <lyandy@google.com> 1601571301 -0700
committer TensorFlower Gardener <gardener@tensorflow.org> 1601571922 -0700

Update TensorFlow MLIR ODS tf.FusedBatchNormGradV3 and tf.FusedBatchNormV3 to support NDHWC and NCDHW data formats.

This is in preparation of https://github.com/tensorflow/tensorflow/pull/42970 where FusedBatchNormGradV3 and FusedBatchNormV3 op defs are updated to support 5D data formats.

PiperOrigin-RevId: 334835832
Change-Id: I04396665beacedd7358ab169606f01b996b00389