def test_softmax(): ncc.clear() ncc.save_tflite(module) init_values() ncc.compile(['--inference-type', 'float']) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 0)
def test_simple_conv2d(): ncc.clear() ncc.save_tflite(module) init_values() ncc.compile(['--inference-type', 'float', '-t', 'cpu']) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 1e-4)
def test_simple_matmul_quant(): ncc.clear() ncc.save_tflite(module) init_values() ncc.compile(['--inference-type', 'uint8', '-t', 'cpu', '--dataset', ncc.input_dir + '/test.bin', '--dataset-format', 'raw', '--input-type', 'float']) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 1e-3)
def test_simple_conv2d_k210(): ncc.clear() ncc.save_tflite(module) init_values() ncc.compile([ '--inference-type', 'uint8', '-t', 'k210', '--dataset', ncc.input_dir + '/test.bin', '--dataset-format', 'raw', '--input-type', 'float' ]) ncc.infer(['--dataset-format', 'raw']) ncc.close_to('test', 0.4)