U qh@@s4ddlmZddlZddlmZGdddeZdS)) annotationsN)OpRunc@seZdZddZdddZdS)ConstantOfShapecCst|||t|jtjr&|jdn|j|_t|jtrJt|j|_n2t|jt rft |j|_n|jdkr|t d|_dS)Nr) r__init__ isinstancevaluenpZndarraycstintZint64floatZfloat64Zfloat32)selfZ onnx_nodeZ run_paramsr K/tmp/pip-unpacked-wheel-xnis5xre/onnx/reference/ops/op_constant_of_shape.pyr s   zConstantOfShape.__init__Nc Cs`ztt||j}WnBtk rX}z$td|d|jd|d|W5d}~XYnX|fS)Nz%Unable to create a constant of shape z with value z (raw value=z).)rfulltupler TypeError RuntimeError)r datarreser r r_runszConstantOfShape._run)N)__name__ __module__ __qualname__rrr r r rr s r) __future__rZnumpyrZonnx.reference.op_runrrr r r rs