} else if (data_format == "NCDHW") {
      (*grad_outputs)[0] =
          Transpose(scope, (*grad_outputs)[0], {0, 4, 1, 2, 3});
    }
    return scope.status();
  }
}

Status FusedBatchNormV3Grad(const Scope& scope, const Operation& op,
                            const std::vector<Output>& grad_inputs,
                            std::vector<Output>* grad_outputs) {
  return BaseFusedBatchNormGrad(

                               public ::testing::WithParamInterface<
                                   std::tuple<bool, bool, TensorShape>> {};

TEST_P(FusedBatchNormGradTest, FusedBatchNormV3Grad) {
  FusedBatchNormV3::Attrs attrs;
  attrs.is_training_ = std::get<0>(GetParam());
  bool channel_first = std::get<1>(GetParam());
  TensorShape shape = std::get<2>(GetParam());