Searched refs:kerns_shp (Results 1 – 2 of 2) sorted by relevance
| /dports/math/py-theano/Theano-1.0.5/theano/gpuarray/ |
| H A D | dnn.py | 980 def _dnn_gradweight(img, topgrad, kerns_shp, alpha=1, beta=0, out=None, border_mode='valid', subsam… argument
986 kerns_shp = theano.tensor.as_tensor_variable(kerns_shp)
994 … conv_mode=conv_mode, precision=precision, num_groups=num_groups)(kerns_shp)
996 real_out = GpuAllocEmpty(dtype=img.dtype, context_name=ctx_name)(*kerns_shp)
1001 real_out = check(out, theano.tensor.all(theano.tensor.eq(out.shape, kerns_shp)))
1249 def dnn_gradweight(img, topgrad, kerns_shp, border_mode='valid', argument
1255 …return _dnn_gradweight(img, topgrad, kerns_shp, border_mode=border_mode, subsample=subsample, dila…
1259 def dnn_gradweight3d(img, topgrad, kerns_shp, border_mode='valid', argument
1265 return dnn_gradweight(img, topgrad, kerns_shp, border_mode,
|
| /dports/math/py-theano/Theano-1.0.5/theano/gpuarray/tests/ |
| H A D | test_dnn.py | 234 assert_types(dnn.dnn_gradweight(mf, md, kerns_shp=mf.shape,
236 assert_types(dnn.dnn_gradweight(md, mf, kerns_shp=mf.shape,
255 assert_types(dnn.dnn_gradweight3d(mf, md, kerns_shp=mf.shape,
257 assert_types(dnn.dnn_gradweight3d(md, mf, kerns_shp=mf.shape,
|