Searched defs:no_batch_dim (Results 1 – 2 of 2) sorted by relevance
51 auto no_batch_dim = self.dim() == 1 && target.dim() == 0; in TORCH_META_FUNC() local97 auto no_batch_dim = self.dim() == 1 && target.dim() == 0; in TORCH_META_FUNC() local
8294 …error_if_input_channels_is_not_num_features(self, device, instance_norm_cls, no_batch_dim, affine): argument