Searched defs:weight_dtype (Results 1 – 8 of 8) sorted by relevance
222 weight_dtype=torch.int8, argument402 weight_dtype=torch.int8, argument
81 at::ScalarType weight_dtype = linear_w.scalar_type(); in FoldFrozenLinearBatchnorm() local
88 at::ScalarType weight_dtype = conv_w.scalar_type(); in FoldFrozenConvBatchnorm() local
409 weight_dtype, argument
1262 weight_dtype = property(lambda self: self.weight.dtype) variable in ExtensionTypeSpecTest.testCustomizeSpecTest.WeightedTensor1269 def __init__(self, shape, dtype, weight_dtype=dtypes.float32): argument1278 weight_dtype = property(lambda self: self.weight.dtype) variable in ExtensionTypeSpecTest.testCustomizeSpecTest.WeightedTensor.Spec
256 def weight_dtype(qconfig): function
228 def weight_dtype(self) -> Optional[torch.dtype]: member in DTypeConfig
4931 weight_dtype=torch.qint8, argument5016 weight_dtype=torch.qint8, argument