fixed typo bug in tensor.blas

上级 c09be4a5
......@@ -461,8 +461,8 @@ class GemmLocalOptimizer(LocalOptimizer):
@staticmethod
def _is_real_matrix(res):
return res.type in T.float_matrix_types \
and res.broadcastable[0] == False \
and res.broadcastable[1] == False #cope with tuple vs. list
and res.type.broadcastable[0] == False \
and res.type.broadcastable[1] == False #cope with tuple vs. list
@staticmethod
def _as_isolated_scalar_times_matrix(res):
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论