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

Go to the source code of this file.

Namespaces

namespace  gen_h5_explicit_inputs_all
 

Functions

 gen_h5_explicit_inputs_all.chunks (lst, n)
 
 gen_h5_explicit_inputs_all.toNumpyType (circle_type)
 

Variables

 gen_h5_explicit_inputs_all.parser = argparse.ArgumentParser()
 
 gen_h5_explicit_inputs_all.type
 
 gen_h5_explicit_inputs_all.required
 
 gen_h5_explicit_inputs_all.help
 
 gen_h5_explicit_inputs_all.nargs
 
 gen_h5_explicit_inputs_all.action
 
 gen_h5_explicit_inputs_all.str
 
 gen_h5_explicit_inputs_all.default
 
 gen_h5_explicit_inputs_all.args = parser.parse_args()
 
 gen_h5_explicit_inputs_all.output_dir = args.output_dir
 
 gen_h5_explicit_inputs_all.artifact_dir = args.artifact_dir
 
 gen_h5_explicit_inputs_all.input_dir = args.input_dir
 
 gen_h5_explicit_inputs_all.test_param = args.test_param
 
 gen_h5_explicit_inputs_all.config = args.config
 
 gen_h5_explicit_inputs_all.mode = args.mode
 
dict gen_h5_explicit_inputs_all.modes_to_input_h5_suffix
 
 gen_h5_explicit_inputs_all.test_params = test_param[0].split()
 
int gen_h5_explicit_inputs_all.PARAM_SET_SIZE = 3
 
 gen_h5_explicit_inputs_all.model_name = test_params[idx][0]
 
 gen_h5_explicit_inputs_all.granularity = test_params[idx][1]
 
 gen_h5_explicit_inputs_all.dtype = test_params[idx][2]
 
 gen_h5_explicit_inputs_all.model = os.path.join(artifact_dir, model_name + '.circle')
 
 gen_h5_explicit_inputs_all.buf = f.read()
 
 gen_h5_explicit_inputs_all.circle_model = Model.GetRootAsModel(buf, 0)
 
 gen_h5_explicit_inputs_all.graph = circle_model.Subgraphs(0)
 
 gen_h5_explicit_inputs_all.inputs = graph.InputsAsNumpy()
 
str gen_h5_explicit_inputs_all.testcase = f'{model_name}.{granularity}.{dtype}'
 
 gen_h5_explicit_inputs_all.output = os.path.join(output_dir, f'{testcase}.{modes_to_input_h5_suffix[mode]}')
 
 gen_h5_explicit_inputs_all.h5_file = h5.File(output, 'w')
 
 gen_h5_explicit_inputs_all.group = h5_file.create_group("value")
 
 gen_h5_explicit_inputs_all.input_text_dir
 
 gen_h5_explicit_inputs_all.records = sorted(glob.glob(input_text_dir + "/*.txt"))
 
 gen_h5_explicit_inputs_all.sample = group.create_group(str(i))
 
 gen_h5_explicit_inputs_all.lines = f.readlines()
 
 gen_h5_explicit_inputs_all.data = np.array(line.split(','))
 
 gen_h5_explicit_inputs_all.input_index = inputs[j]
 
 gen_h5_explicit_inputs_all.tensor = graph.Tensors(input_index)
 
 gen_h5_explicit_inputs_all.np_type = toNumpyType(tensor.Type())
 
 gen_h5_explicit_inputs_all.input_data = np.array(data.reshape(tensor.ShapeAsNumpy()), np_type)