./aten/src/ATen/nn.yaml is also updated. But still the error is
ten/build/src/ATen/ATen/CUDADoubleType.cpp: In member function ‘virtual std::tuple<at::Tensor, at::Tensor, at::Tensor> at::CUDADoubleType::thnn_batch_norm_forward(const at::Tensor&, const at::Tensor&, const at::Tensor&, const at::Tensor&, const at::Tensor&, bool, double, double) const’: