提交 7330be54 authored 作者: Arnaud Bergeron's avatar Arnaud Bergeron

Make the random generator work.

上级 2f986160
......@@ -771,6 +771,9 @@ class GPUA_mrg_uniform(GpuKernelBase, mrg_uniform_base):
# GpuArray version
_f16_ok = True
def get_context(self, node):
return node.inputs[0].type.context
@classmethod
def new(cls, rstate, ndim, dtype, size):
v_size = as_tensor_variable(size)
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论