3e5f67ba78d7bdfd64735a0679a488613b1b6fc9,tests/conversion/onnx_layer_test_cases.py,ONNXLayerTestCases,test_ArgMin,#ONNXLayerTestCases#,74

Before Change

        return model.net.Proto(), model.param_init_net.Proto(), value_info

    def test_ArgMin(self):
        self._test_from_caffe2(self.ArgMin_network)

    // def test_Asin(self):
    //     pass
    //

After Change

        g.outputs = (y,)
        self._test(g)

    def test_ArgMin(self):
        g = ONNXGraph(self._graph_name("ArgMin"))
        x = ONNXTensor(graph=g, name="x", shape=[1, 5, 5, 5], dtype="FLOAT")
        y = ONNXTensor(graph=g, name="y", shape=[1, 1, 5, 5], dtype="INT64")
        ONNXOperation(graph=g, name="ArgMin", inputs=(x,), outputs=(y,), attribs=dict(axis=1, keepdims=1))
        g.inputs = (x,)
        g.outputs = (y,)
        self._test(g)

    // def test_Asin(self):
    //     pass
    //

Instances