CMSIS-NN: Implement concatenation layer - Int8\Uint8

pull/19/head
Gian Marco Iodice 6 years ago committed by felix-johnny
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…
Cancel
Save