def test_constant():
c = constant(2, name="a")
assert c.name == "a"
assert c.dtype == "int8"
c = constant(2, dtype="float32")
assert c.name is None
assert c.dtype == "float32"
def test_composite_clone_float32(self):
def has_f16(comp):
if any(v.type == float16 for v in comp.fgraph.variables):
return True
return False
w = int8()
x = float16()
y = float32()
cz = Composite([x, y], [tanh(x + cast(y, "float16"))])
c = Composite(
[w, x, y],
[
cz(x, y) - cz(x, y)**2 + cast(x, "int16") +
cast(x, "float32") + cast(w, "float16") -
constant(np.float16(1.0))
],
)
assert has_f16(c)
nc = c.clone_float32()
assert not has_f16(nc)
v = uint8()
w = float16()
x = float16()
y = float16()
z = float16()
c = Composite([v, w, x, y, z], [switch(v, mul(w, x, y), z)])
assert has_f16(c)
nc = c.clone_float32()
assert not has_f16(nc)