Searched refs:BatchNorm1d (Results 1 – 7 of 7) sorted by relevance
23 self.bn_0 = nn.BatchNorm1d(num_features=32)24 self.bn_1 = nn.BatchNorm1d(num_features=32, eps=1e-1, affine=False)25 self.bn_2 = nn.BatchNorm1d(num_features=11, affine=True)
24 self.bn_0 = nn.BatchNorm1d(num_features=16)26 self.bn_1 = nn.BatchNorm1d(num_features=3)
21 class BatchNorm1d : public FuseModulePass class55 REGISTER_GLOBAL_PNNX_FUSE_MODULE_PASS(BatchNorm1d)
10 class NaiveSyncBatchNorm1d(nn.BatchNorm1d):
14 torch.nn.BatchNorm1d,
427 |nn.BatchNorm1d | :heavy_check_mark: | :heavy_check_mark: |