Searched full:batch_norm2d (Results 1 – 10 of 10) sorted by relevance
89 return torch.ops.quantized.batch_norm2d(
120 self.batch_norm2d = torch.nn.BatchNorm2d(3, affine=False)128 x = self.batch_norm2d(x)
74 return torch.ops.quantized.batch_norm2d(
409 …m.impl(TORCH_SELECTIVE_NAME("quantized::batch_norm2d"), TORCH_FN(q_batch_norm2d_impl<false>)); in TORCH_LIBRARY_IMPL()
419 quantized::batch_norm2d: 1
665 - quantized::batch_norm2d1028 quantized::batch_norm2d: 1
488 "batch_norm2d.default": _transform_batch_norm,
317 exir_ops.edge.quantized.batch_norm2d.default,
51 …m.def(TORCH_SELECTIVE_SCHEMA("quantized::batch_norm2d(Tensor qx, Tensor? weight, Tensor? bias, Ten… in TORCH_LIBRARY()
2752 qy = torch.ops.quantized.batch_norm2d(