MCPcopy Create free account
hub / github.com/apache/tvm / verify

Function verify

tests/python/nightly/test_nnapi/test_ops.py:352–357  ·  view source on GitHub ↗
(remote_obj, tracker, mod, inputs)

Source from the content-addressed store, hash-verified

350
351
352def verify(remote_obj, tracker, mod, inputs):
353 inputs_tvm: list[tvm.runtime.Tensor] = [tvm.runtime.tensor(v) for v in inputs]
354 outputs = _build_and_run_network(remote_obj, tracker, mod, inputs_tvm)
355 nnapi_out = outputs[0]
356 expected_out = outputs[1]
357 tvm.testing.assert_allclose(nnapi_out, expected_out, rtol=1e-4, atol=1e-5)
358
359
360if __name__ == "__main__":

Callers 9

test_unaryFunction · 0.70
test_elementwise_binaryFunction · 0.70
test_divideFunction · 0.70
test_matmulFunction · 0.70
test_permute_dimsFunction · 0.70
test_astypeFunction · 0.70
test_meanFunction · 0.70
test_conv2dFunction · 0.70
test_max_pool2dFunction · 0.70

Calls 1

_build_and_run_networkFunction · 0.70

Tested by

no test coverage detected

Used in the wild real call sites across dependent graphs

searching dependent graphs…