Skip to content
Snippets Groups Projects
Commit 13c6f600 authored by auphelia's avatar auphelia
Browse files

[Code generation] Added StreamingFCLayer_Batch class to layers.py and its init function

parent b2a6d571
No related branches found
No related tags found
No related merge requests found
class StreamingFCLayer_Batch():
def __init__(self, node, model):
# Layer attributes
num_attr = len(node.attribute)
for k in range(num_attr):
if node.attribute[k].name == "PE":
self.PE = node.attribute[k].i
if node.attribute[k].name == "SIMD":
self.SIMD = node.attribute[k].i
if node.attribute[k].name == "MH":
self.MH = node.attribute[k].i
if node.attribute[k].name == "MW":
self.MW = node.attribute[k].i
if node.attribute[k].name == "resDataType":
self.resDataType = node.attribute[k].s.decode("utf-8")
if node.attribute[k].name == "resType":
self.resType = node.attribute[k].s.decode("utf-8")
# get other parameters
weights_shape = model.get_tensor_shape(node.input[1])
thresholds_shape = model.get_tensor_shape(node.input[2])
self.WMEM = weights_shape[2]
self.TMEM = thresholds_shape[0]
self.API = thresholds_shape[2]
def get_PE(self):
return self.PE
def get_SIMD(self):
return self.SIMD
def get_MH(self):
return self.MH
def get_MW(self):
return self.MW
def get_resDataType(self):
return self.resDataType
def get_resType(self):
return self.resType
def get_WMEM(self):
return self.WMEM
def get_TMEM(self):
return self.TMEM
def get_API(self):
return self.API
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment