ONE - On-device Neural Engine
Loading...
Searching...
No Matches
compare_tensors_all.py File Reference

Go to the source code of this file.

Namespaces

namespace  compare_tensors_all
 

Functions

 compare_tensors_all.compare_fake_quantization (tensor, tensor_name, expect_dir)
 
 compare_tensors_all.compare_record_minmax (tensor, tensor_name, expect_dir)
 
 compare_tensors_all.compare_quantization (tensor, tensor_name, expect_dir)
 
 compare_tensors_all.chunks (lst, n)
 

Variables

 compare_tensors_all.parser = argparse.ArgumentParser()
 
 compare_tensors_all.type
 
 compare_tensors_all.required
 
 compare_tensors_all.nargs
 
 compare_tensors_all.help
 
 compare_tensors_all.str
 
 compare_tensors_all.True
 
 compare_tensors_all.args = parser.parse_args()
 
dict compare_tensors_all.modes_to_expected_folder
 
dict compare_tensors_all.modes_to_input_h5_suffix
 
 compare_tensors_all.test_param = args.test_param
 
 compare_tensors_all.bin_dir = args.bin_dir
 
 compare_tensors_all.source_dir = args.source_dir
 
 compare_tensors_all.mode = args.mode
 
str compare_tensors_all.log_format = '%(levelname)s: %(message)s'
 
 compare_tensors_all.formatter = logging.Formatter(log_format)
 
 compare_tensors_all.streamer = StringIO()
 
 compare_tensors_all.stream_handler = logging.StreamHandler(stream=streamer)
 
 compare_tensors_all.handlers
 
 compare_tensors_all.failed_log = dict()
 
 compare_tensors_all.inputs = test_param[0].split()
 
int compare_tensors_all.PARAM_SET_SIZE = 3
 
 compare_tensors_all.model_name = inputs[idx][0]
 
 compare_tensors_all.granularity = inputs[idx][1]
 
 compare_tensors_all.dtype = inputs[idx][2]
 
str compare_tensors_all.testcase = f'{model_name}.{granularity}.{dtype}'
 
 compare_tensors_all.test_result_file = os.path.join(bin_dir, testcase)
 
str compare_tensors_all.input_h5 = f'{test_result_file}.{modes_to_input_h5_suffix[mode]}'
 
str compare_tensors_all.expect_dir = f'{source_dir}/expected_outputs/{model_name}/{granularity}/{dtype}/{modes_to_expected_folder[mode]}'
 
bool compare_tensors_all.test_result = False
 
 compare_tensors_all.failed_number = len(failed_log)