提交 1d69eac9 authored 作者: Frédéric Bastien's avatar Frédéric Bastien

Merge pull request #3513 from mronian/master

conv2d support uint
......@@ -108,7 +108,7 @@ def conv2d(input, filters, image_shape=None, filter_shape=None,
" information are constant values. We got"
" %s for the image_shape parameter" %
image_shape[i])
assert str(image_shape[i].dtype).startswith('int')
assert "int" in str(image_shape[i].dtype)
image_shape[i] = int(image_shape[i])
if filter_shape is not None:
filter_shape = list(filter_shape)
......@@ -123,7 +123,7 @@ def conv2d(input, filters, image_shape=None, filter_shape=None,
" information are constant values. We got"
" %s for the filter_shape "
"parameter" % filter_shape[i])
assert str(filter_shape[i].dtype).startswith('int')
assert "int" in str(filter_shape[i].dtype)
filter_shape[i] = int(filter_shape[i])
if image_shape and filter_shape:
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论