3j&ddlmZddlZ d dd ZdS)) annotationsNvalue np.ndarrayshapetuple[int, ...]axisint block_size int | Nonereturnc t|jdkr|St|jdkr|jdkr|dS|st|jdksJdgt|z} |j||<|t |S#t $r$}t d|d|jd|d|d}~wwxYw|dkrt dtj||| }|||j|kr*| t||| }|j|krt d tj ||j|ksJ|S) aReshape/Replicate scale/zero-point to be broadcastable to shape. Args: value: the array to be reshaped/replicated shape: the target shape axis: quantization axis, applicable for per-axis and blocked quantization block_size: size of quantization block, applicable only for blocked quantization Returns: value array after reshape/replicate according to quantization mode. rzaxis is out of boundary, axis=z, value.shape=z, shape=.Nz&block_size must be a positive integer.)repeatsr)indicesrzInvalid shapes for Blocked Quantization. Input 2 shape should identical to Input 1 shape, except for one dimension, in which blocking is performed) lenrsizereshapetuple IndexError ValueErrornprepeattakerangebroadcast_shapes)rrrr dimses PC:\Projects\Historical\venv\Lib\site-packages\onnx/reference/ops/_quant_utils.py reshape_inputr s" 5;1  5;! aQx  5;1$$$$sSZZ DJ==t-- -   ===${==49===   QABBB IeZd ; ; ;E d u{4((( 5t#5#5D AA {e a     uek 2 2e ; ; ; ; Ls6+B"" C,C  C)N) rrrrrr r r r r) __future__rnumpyrr rr%sM#"""""" ///////r$