def test_op(self): attrs = framework.NodeAttr() attrs.attr_store = { "scale": 0.6, "bias": 0.4, "bias_after_scale": False } self.to_test_op([[100, 32]], [[100, 32]], "scale", attrs)
def test_op(self): attrs = framework.NodeAttr() attrs.attr_store = {"padding": [1, 1]} attrs.set_attr("stride", [2, 2]) attrs.set_attr("dilation", [2, 2]) attrs.set_attr("groups", 1) self.to_test_op([[1, 3, 10, 10], [2, 3, 2, 2]], [[1, 3, 12, 12], [2, 3, 3, 3], [1, 2, 5, 5]], "conv2d", attrs)
class OpTest_pool1d_1(SingleOpTester): attrs = framework.NodeAttr() attrs.attr_store = { "kernel_size": [2], "stride_size": [2], "padding_size": [2, 3], "pool_type": "avg", "ceil_mode": False, "exclusive": True, "data_format": "NCW" } def create_target_data(self, inputs_data, attrs): return pool_utils.pool1d(inputs_data[0], self.attrs) def test_op(self): input_shape = [1, 3, 8] self.to_test_op([input_shape], None, "pool1d", self.attrs)
def test_op(self): attrs = framework.NodeAttr() attrs.attr_store = {"scale": 0.7, "bias": 0.3} self.to_test_op([[100, 32]], [[100, 32]], "scale", attrs)
def test_op(self): attrs = framework.NodeAttr() attrs.attr_store = {"axis": 1} self.to_test_op([[3, 2], [2]], [[3, 2]], "elementwise_mul", attrs)
def test_op(self): attrs = framework.NodeAttr() attrs.attr_store = {"axis": 0} self.to_test_op([[100, 32], [100, 32]], [[100, 32]], "elementwise_add", attrs)
def test_op(self): attrs = framework.NodeAttr() self.to_test_op([[3, 224, 224]], [[3, 224, 224]], "sigmoid", attrs)
def test_op(self): attrs = framework.NodeAttr() attrs.set_attr("axis", 0) self.to_test_op([[12, 224, 224]], [[12, 224, 224], [12, 224, 224]], "softmax", attrs)
def test_op(self): attrs = framework.NodeAttr() self.to_test_op([[1, 64, 112, 112], [64], [64], [64], [64]], [[1, 64, 112, 112]], "batchnorm", attrs)
def test_op(self): attrs = framework.NodeAttr() self.to_test_op([[32, 32]], [[32, 32]], "relu6", attrs)