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)
