()
| 169 | |
| 170 | @requires_coreml_runtime |
| 171 | def test_clip(): |
| 172 | x = relax.Var("x", relax.TensorStructInfo([10, 10], "float32")) |
| 173 | bb = relax.BlockBuilder() |
| 174 | |
| 175 | with bb.function("main", [x]): |
| 176 | with bb.dataflow(): |
| 177 | lv0 = bb.emit(relax.op.clip(x, 0, 4)) |
| 178 | gv0 = bb.emit_output(lv0) |
| 179 | bb.emit_func_output(gv0) |
| 180 | mod = bb.get() |
| 181 | |
| 182 | x_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev) |
| 183 | verify(mod, [x_data]) |
| 184 | |
| 185 | x = relax.Var("x", relax.TensorStructInfo([10, 10], "float32")) |
| 186 | bb = relax.BlockBuilder() |
| 187 | |
| 188 | with bb.function("main", [x]): |
| 189 | with bb.dataflow(): |
| 190 | lv0 = bb.emit(relax.op.clip(x, 0, 4)) |
| 191 | lv1 = bb.emit(relax.op.clip(x, 1, 3)) |
| 192 | gv0 = bb.emit_output(lv0) |
| 193 | gv1 = bb.emit_output(lv1) |
| 194 | bb.emit_func_output([gv0, gv1]) |
| 195 | |
| 196 | x_data = tvm.runtime.tensor(np.random.rand(10, 10).astype("float32"), dev) |
| 197 | verify(mod, [x_data]) |
| 198 | |
| 199 | |
| 200 | @requires_coreml_runtime |
nothing calls this directly
no test coverage detected
searching dependent graphs…