o r::js@s&ddlmZddlZ ddd d ZdS)) annotationsNvalue np.ndarrayshapetuple[int, ...]axisint block_size int | Nonereturnc Cst|jdkr |St|jdkr|jdkr|dS|sUt|jdks$Jdgt|}z |j||<|t|WStyT}ztd|d|jd|d|d}~ww|dkr]tdtj|||d }|||j|kry|j t |||d }|j|krtd t ||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 dimserf/home/nk/hobo-godmode/plappi-mvp/.venv/lib/python3.10/site-packages/onnx/reference/ops/_quant_utils.py reshape_input sB  r)N) rrrrrrr r r r) __future__rnumpyrrrrrrs