__init__.py

Go to the documentation of this file.

from test_utils import TestCase
from test_utils import gen_random_tensor
 
 
class AveragePool2D_000_Q8(TestCase):
    def __init__(self):
        self.name = _name_
 
    def generate(self) -> dict:
        json_content = dict()
 
        # Generate ifm
        json_content['ifm'] = gen_random_tensor(
            "uint8",  # dtype_str
            (1),  # scale_shape
            (1),  # zerop_shape
            0)  # quantized_dimension
 
        # Generate ofm
        json_content['ofm'] = gen_random_tensor(
            "uint8",  # dtype_str
            (1),  # scale_shape
            (1),  # zerop_shape
            0)  # quantized_dimension
 
        return json_content
 
 
_name_ = 'AveragePool2D_000_Q8'
 
_test_case_ = AveragePool2D_000_Q8()