提交 4c326058 authored 作者: Frederic's avatar Frederic

Make all dnn op do not check the broadcast flags.

上级 574753c0
......@@ -88,6 +88,10 @@ class DnnBase(GpuOp):
"""
Creates a handle for cudnn and pulls in the cudnn libraries and headers.
"""
# dnn do know know about broadcasting, so we do not need to assert
# the input broadcasting pattern.
check_broadcast = False
def c_headers(self):
return ['cudnn.h', 'cudnn_helper.h']
......@@ -244,7 +248,6 @@ class GpuDnnConvDesc(GpuOp):
class GpuDnnConvBase(DnnBase):
__props__ = ()
check_broadcast = False
def c_support_code_struct(self, node, struct_id):
return """
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论