Searched defs:bn_module (Results 1 – 2 of 2) sorted by relevance
378 bn_module=None, # For BN Module argument
4840 def test_batchnorm_non_contig_cpu(self, bn_module): argument