1 namespace at::native { 2 3 TORCH_API size_t 4 _get_cudnn_batch_norm_reserve_space_size(const Tensor& input_t, bool training); 5 6 } // namespace at::native 7