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')
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')
