def test_ArgMin(tmpdir, dtype): with C.default_options(dtype = dtype): shape = (4, 5) data = np.random.rand(*shape).astype(dtype) model = C.argmin(data, 0) verify_no_input(model, tmpdir, 'ArgMin_0')
def test_ArgMin(tmpdir, dtype): with C.default_options(dtype=dtype): shape = (4, 5) data = np.random.rand(*shape).astype(dtype) model = C.argmin(data, 0) verify_no_input(model, tmpdir, 'ArgMin_0')
def test_ArgMin(tmpdir): shape = (4, 5) data = np.random.rand(*shape).astype(np.float32) model = C.argmin(data, 0) verify_no_input(model, tmpdir, 'ArgMin_0')