CMSIS-NN: Implement concatenation layer - Int8\Uint8
parent
bae42f246c
commit
a9d1c1cb0c
@ -1,18 +1,18 @@
|
|||||||
import argparse
|
import argparse
|
||||||
import TestScripts.NewParser as parse
|
import TestScripts.NewParser as parse
|
||||||
import pickle
|
import pickle
|
||||||
|
|
||||||
parser = argparse.ArgumentParser(description='Parse test description')
|
parser = argparse.ArgumentParser(description='Parse test description')
|
||||||
|
|
||||||
parser.add_argument('-f', nargs='?',type = str, default=None, help="Test description file path")
|
parser.add_argument('-f', nargs='?',type = str, default=None, help="Test description file path")
|
||||||
|
|
||||||
parser.add_argument('-o', nargs='?',type = str, default="Output.pickle", help="output file for parsed description")
|
parser.add_argument('-o', nargs='?',type = str, default="Output.pickle", help="output file for parsed description")
|
||||||
|
|
||||||
args = parser.parse_args()
|
args = parser.parse_args()
|
||||||
|
|
||||||
if args.f is not None:
|
if args.f is not None:
|
||||||
p = parse.Parser()
|
p = parse.Parser()
|
||||||
# Parse the test description file
|
# Parse the test description file
|
||||||
root = p.parse(args.f)
|
root = p.parse(args.f)
|
||||||
with open(args.o,"wb") as output:
|
with open(args.o,"wb") as output:
|
||||||
pickle.dump(root, output)
|
pickle.dump(root, output)
|
||||||
|
|||||||
Loading…
Reference in New Issue