get_block_size
-
class torch.ao.quantization.observer.get_block_size(input_shape, granularity)[source] -
Get the block size based on the input shape and granularity type.
- Parameters
-
- input_shape (tuple[int, ...]) – The input tensor shape possibly more than 2 dimensions
- granularity (Granularity) – The granularity type of the quantization
- Return type