提交 7141f49d authored 作者: sentient07's avatar sentient07

Changed the condition for first input

上级 a0350219
......@@ -883,9 +883,8 @@ def local_gpua_softmaxwithbias(node, context_name):
def local_assert(node, context_name):
# Check if input nodes are already on the GPU
for i in node.inputs:
if isinstance(i.type, GpuArrayType):
return
if isinstance(node.inputs[0].type, GpuArrayType):
return
return [host_from_gpu(node.op(as_gpuarray_variable(node.inputs[0],
context_name),
*node.inputs[1:]))]
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论