D | batchnorm_test.py | 35 "training": [True, False], 47 training=[True, False], 54 def init(self, M, N, K, device, training, cudnn): argument 63 "training": training, 68 def forward(self, input_one, mean, var, weight, bias, training, cudnn): argument 70 return F.batch_norm(input_one, mean, var, weight, bias, training) 89 "training": [True, False], 100 training=[True, False], 107 def init(self, N, C, device, training, cudnn): argument 114 "training": training, [all …]
|