3e5f67ba78d7bdfd64735a0679a488613b1b6fc9,tests/conversion/onnx_layer_test_cases.py,ONNXLayerTestCases,test_ReduceLogSumExp,#ONNXLayerTestCases#,642

Before Change

        self._test_from_onnx_graph(g, "ReduceLogSum", run=False)

    def test_ReduceLogSumExp(self):
        g = ONNXGraph("test_network")
        x = ONNXTensor(graph=g, name="x", shape=[1, 2, 5, 5], dtype="FLOAT")
        y = ONNXTensor(graph=g, name="y", shape=[1, 2, 1, 1], dtype="FLOAT")
        ONNXOperation(graph=g, name="ReduceLogSumExp",
                      inputs=x,
                      attribs=dict(axes=[2, 3]),
                      outputs=y)
        g.inputs = (x,)
        g.outputs = (y,)
        self._test_from_onnx_graph(g, "ReduceLogSumExp", run=False)

    def test_ReduceMax(self):
        self._test_from_caffe2(self.get_unary_network_function("ReduceMax", kwargs=dict(axes=[1, 2])))
        self._test_from_caffe2(self.get_unary_network_function("ReduceMax", kwargs=dict(axes=[1, 2], keepdims=0)))

After Change

        self._test(self._unary("ReduceLogSum", out_shape=[1, 5], attribs=dict(axes=[1, 2], keepdims=0)), run=False)

    def test_ReduceLogSumExp(self):
        self._test(self._unary("ReduceLogSumExp", out_shape=[1, 1, 1, 5], attribs=dict(axes=[1, 2])), run=False)
        self._test(self._unary("ReduceLogSumExp", out_shape=[1, 5], attribs=dict(axes=[1, 2], keepdims=0)), run=False)

    def test_ReduceMax(self):
        self._test(self._unary("ReduceMax", out_shape=[1, 1, 1, 5], attribs=dict(axes=[1, 2])))
        self._test(self._unary("ReduceMax", out_shape=[1, 5], attribs=dict(axes=[1, 2], keepdims=0)))

Instances