o i@sddlmZmZddlZddlmZmZmZm Z ddl m Z e dee efdee efdefd d Z e dee efdee efdefd d Ze dee efdee efdefd dZe dedede fddZe deeeee fefdejdefddZdS))UnionTupleN) ShapeExpr TensorDesc ShapeExprsSizeTensorDesc) public_apifirstsecondreturncC4t|trt|trtdt|}|tjj|S)ap Computes symbolic ceiling division of `first` by `second` Args: first (Union[int, ShapeExpr]): Dividend second (Union[int, ShapeExpr]): Divisor Raises: ValueError: If both arguments are `int`\s or if `second` evaluates to 0 Returns: ShapeExpr: Symbolic expression for the ceiling division of `first` by `second` Both arguments cannot be 'int's) isinstanceint ValueErrorr_optrtDimensionOperationCEIL_DIVr r rW/home/ubuntu/.local/lib/python3.10/site-packages/tensorrt_bindings/plugin/_top_level.pycdiv  rcCr )aI Computes the maximum of `first` and `second` Args: first (Union[int, ShapeExpr]): First operand second (Union[int, ShapeExpr]): Second operand Raises: ValueError: If both arguments are `int`\s Returns: ShapeExpr: Symbolic expression for the maximum of `first` and `second` r)rrrrrrrMAXrrrrmax2rrcCr )aI Computes the minimum of `first` and `second` Args: first (Union[int, ShapeExpr]): First operand second (Union[int, ShapeExpr]): Second operand Raises: ValueError: If both arguments are `int`\s Returns: ShapeExpr: Symbolic expression for the minimum of `first` and `second` r)rrrrrrrMINrrrrminJrropt upper_boundcCs t||S)a Constructs a size tensor with the specified autotune shape expression `opt` and `upper_bound` Args: opt (ShapeExpr): Symbolic expression for the extent of this size tensor to use in the autotune process of the engine build upper_bound (ShapeExpr): Symbolic expression for the upper-bound of this size tensor Returns: SizeTensorDesc: A tensor descriptor for a size tensor with the specified autotune extent and upper-bound )r)rr rrr size_tensorbs r! shape_exprdtypecCs$t|tr t|}n|}t||S)a Constructs a tensor descriptor with the specified shape expression and data type Args: shape_expr (Union[Tuple[Union[ShapeExpr, int]], ShapeExprs]): Expressions or constants denoting the shape of the tensor dtype (trt.DataType): Data type of the tensor Returns: TensorDesc: Tensor descriptor with the specified shape expression and data type )rtupler from_tupler)r"r# shape_expr_rrrfrom_shape_exprqs  r')typingrrtensorrtr_tensorrrrr_exportr rrrrr!DataTyper'rrrrs (((2