提交 887947e7 authored 作者: James Bergstra's avatar James Bergstra

added handle_shared_float32 function to package namespace

上级 e8685b15
......@@ -7,3 +7,14 @@ from .var import (CudaNdarrayVariable,
import basic_ops
import opt
import theano.compile.sandbox
def handle_shared_float32(tf):
"""Set the CudaNdarrayType as the default handler for shared float32 arrays
"""
if tf:
theano.compile.sandbox.shared_constructor(shared_constructor)
else:
raise NotImplementedError('removing our handler')
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论