提交 ea493492 authored 作者: Samira Shabanian's avatar Samira Shabanian

new changes

上级 6319c6b4
...@@ -762,8 +762,6 @@ class CSM(gof.Op): ...@@ -762,8 +762,6 @@ class CSM(gof.Op):
if self.kmap is None: if self.kmap is None:
# node.inputs[3] is of lenght as we only support sparse matrix. # node.inputs[3] is of lenght as we only support sparse matrix.
return [(node.inputs[3][0], node.inputs[3][1])] return [(node.inputs[3][0], node.inputs[3][1])]
else:
raise Exception("Do not use kmap, it is removed")
CSC = CSM('csc') CSC = CSM('csc')
""" """
...@@ -896,8 +894,6 @@ class CSMGrad(gof.op.Op): ...@@ -896,8 +894,6 @@ class CSMGrad(gof.op.Op):
def infer_shape(self, node, shapes): def infer_shape(self, node, shapes):
if self.kmap is None: if self.kmap is None:
return [shapes[1]] return [shapes[1]]
else:
raise Exception("Do not use kmap, it is removed")
csm_grad = CSMGrad csm_grad = CSMGrad
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论