Skip to content
Snippets Groups Projects
Commit 87529826 authored by Yaman Umuroglu's avatar Yaman Umuroglu
Browse files

[InferVVAU] add suport for setting mem_mode at conversion time

parent e5114ef0
No related branches found
No related tags found
No related merge requests found
......@@ -870,6 +870,10 @@ class InferVectorVectorActivation(Transformation):
a depthwise convolution. Any immediately following MultiThreshold
layers will also be absorbed into the VVAU."""
def __init__(self, mem_mode="const"):
super().__init__()
self.mem_mode = mem_mode
def apply(self, model):
graph = model.graph
node_ind = 0
......@@ -970,6 +974,7 @@ class InferVectorVectorActivation(Transformation):
ActVal=actval,
noActivation=0,
name="VectorVectorActivation_" + n.name,
mem_mode=self.mem_mode,
)
graph.node.insert(node_ind, new_node)
# remove old nodes
......
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