|
ONE - On-device Neural Engine
|
Variables | |
| parser = argparse.ArgumentParser() | |
| type | |
| int | |
| required | |
| str | |
| True | |
| nargs | |
| args = parser.parse_args() | |
| num_data = args.num_data | |
| output_dir = args.output_dir | |
| artifact_dir = args.artifact_dir | |
| model_list = args.model | |
| model_path = os.path.join(artifact_dir, model_name + '.tflite') | |
| h5_path = os.path.join(output_dir, model_name + '.tflite.input.h5') | |
| interpreter = tf.lite.Interpreter(model_path) | |
| input_details = interpreter.get_input_details() | |
| h5_file = h5.File(h5_path, 'w') | |
| group = h5_file.create_group("value") | |
| sample = group.create_group(str(i)) | |
| input_detail = input_details[j] | |
| input_data | |
| data | |
| gen_h5_random_inputs_all.args = parser.parse_args() |
Definition at line 16 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.artifact_dir = args.artifact_dir |
Definition at line 20 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.data |
Definition at line 53 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.group = h5_file.create_group("value") |
Definition at line 32 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.h5_file = h5.File(h5_path, 'w') |
Definition at line 31 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.h5_path = os.path.join(output_dir, model_name + '.tflite.input.h5') |
Definition at line 25 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.input_data |
Definition at line 45 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.input_detail = input_details[j] |
Definition at line 41 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.input_details = interpreter.get_input_details() |
Definition at line 28 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.int |
Definition at line 12 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.interpreter = tf.lite.Interpreter(model_path) |
Definition at line 27 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.model_list = args.model |
Definition at line 21 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.model_path = os.path.join(artifact_dir, model_name + '.tflite') |
Definition at line 24 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.nargs |
Definition at line 15 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.num_data = args.num_data |
Definition at line 18 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.output_dir = args.output_dir |
Definition at line 19 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.parser = argparse.ArgumentParser() |
Definition at line 11 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.required |
Definition at line 12 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.sample = group.create_group(str(i)) |
Definition at line 37 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.str |
Definition at line 13 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.True |
Definition at line 15 of file gen_h5_random_inputs_all.py.
| gen_h5_random_inputs_all.type |
Definition at line 12 of file gen_h5_random_inputs_all.py.