ONE - On-device Neural Engine
Loading...
Searching...
No Matches
Namespaces
|
Functions
|
Variables
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
)
compiler
pota-quantization-value-test
gen_h5_explicit_inputs_all.py
Generated by
1.9.8