Upload 379 files

610d9c2 verified about 3 hours ago

928 Bytes

from .direct_q8 import Direct8BitOp, QDQDirect8BitOp


class QResize(Direct8BitOp):
    def __init__(self, onnx_quantizer, onnx_node):
        super().__init__(onnx_quantizer, onnx_node)

    def quantize(self):
        node = self.node
        assert node.op_type == "Resize"

        # if version is less than 11, go to normal quantize.
        if self.quantizer.opset_version < 11:
            super(Direct8BitOp, self).quantize()
            return

        # Direct 8bits op
        return super().quantize()


class QDQResize(QDQDirect8BitOp):
    def __init__(self, onnx_quantizer, onnx_node):
        super().__init__(onnx_quantizer, onnx_node)

    def quantize(self):
        node = self.node
        assert node.op_type == "Resize"

        # if version is less than 11, just keep this node
        if self.quantizer.opset_version < 11:
            return

        # Direct 8bits op
        return super().quantize()

	from .direct_q8 import Direct8BitOp, QDQDirect8BitOp


	class QResize(Direct8BitOp):
	def __init__(self, onnx_quantizer, onnx_node):
	super().__init__(onnx_quantizer, onnx_node)

	def quantize(self):
	node = self.node
	assert node.op_type == "Resize"

	# if version is less than 11, go to normal quantize.
	if self.quantizer.opset_version < 11:
	super(Direct8BitOp, self).quantize()
	return

	# Direct 8bits op
	return super().quantize()


	class QDQResize(QDQDirect8BitOp):
	def __init__(self, onnx_quantizer, onnx_node):
	super().__init__(onnx_quantizer, onnx_node)

	def quantize(self):
	node = self.node
	assert node.op_type == "Resize"

	# if version is less than 11, just keep this node
	if self.quantizer.opset_version < 11:
	return

	# Direct 8bits op
	return super().quantize()