提交 baa3c014 authored 作者: Ian Goodfellow's avatar Ian Goodfellow

merged

...@@ -340,7 +340,7 @@ class Conv3D(theano.Op): ...@@ -340,7 +340,7 @@ class Conv3D(theano.Op):
codeSource += """ codeSource += """
if (inputChannels > 20 && outputChannels > 20 && ws4 == sizeof(ELEM_AT(%(W)s,0))) if (inputChannels > 20 && outputChannels > 20 && ws4 == sizeof(ELEM_AT(%(W)s,0)))
{ {
std::cout << "lots of channels special case code" << std::endl; //std::cout << "lots of channels special case code" << std::endl;
#define blas_type dtype_ ## %(V)s #define blas_type dtype_ ## %(V)s
const blas_type constant_one = 1.0; const blas_type constant_one = 1.0;
char N = 'T'; char N = 'T';
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论