U ʹj+@sdZddlZddlZddlmZmZddlmZm Z ej e j ddZ dZ e ddejd d d Ze d dejd d dZe D]Ze deeeqdS)a' Note [ONNX operators that are added/updated from opset 7 to opset 8] ~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~~ New operators: Expand Updated operators: Min, Max, Sum, Mean: supports multidirectional broadcasting. MaxPool: added optional indices output. Scan N)symbolic_helpersymbolic_opset9) jit_utils registration)Zopset) scanexpandZ expand_asZmeshgridZadaptive_max_pool1dZadaptive_max_pool2dZadaptive_max_pool3dZmax_pool1d_with_indicesZmax_pool2d_with_indicesZmax_pool3d_with_indicesz aten::max)gcCs*|dkr|dk rtdt||||S)NzMultidirectional broadcasting is not supported in opset 7. This might cause the onnx model to be incorrect, if inputs to max operators have different shapes)warningswarnopset9maxr selfZdim_or_yZkeepdimrY/var/www/html/bwave/python/venv/lib/python3.8/site-packages/torch/onnx/symbolic_opset7.pyr 's r z aten::mincCs*|dkr|dk rtdt||||S)NzMultidirectional broadcasting is not supported in opset 7. This might cause the onnx model to be incorrect, if inputs to min operators have different shapes)r r r minrrrrr3s rzaten::)NN)NN)__doc__ functoolsr Z torch.onnxrrr Ztorch.onnx._internalrrpartialZ onnx_symbolicZ_onnx_symbolicZblock_listed_operatorsZ GraphContextr rZblock_listed_opZ_block_list_in_opsetrrrrs