提交 1b5d0fd0 authored 作者: Thomas George's avatar Thomas George

uses bool from scalar instead of recreating it

上级 1eb701aa
...@@ -7,7 +7,7 @@ from theano import Apply, config, Op ...@@ -7,7 +7,7 @@ from theano import Apply, config, Op
from theano.compile import optdb from theano.compile import optdb
from theano.gof import LocalOptGroup, ParamsType from theano.gof import LocalOptGroup, ParamsType
from theano.scalar import Scalar from theano.scalar import bool as bool_t
from theano.tensor.basic import as_tensor_variable from theano.tensor.basic import as_tensor_variable
from theano.tensor.opt import in2out from theano.tensor.opt import in2out
...@@ -39,7 +39,7 @@ class GpuGemv(BlasOp): ...@@ -39,7 +39,7 @@ class GpuGemv(BlasOp):
Gemv on the GPU. Gemv on the GPU.
""" """
params_type = ParamsType(inplace=Scalar('bool')) params_type = ParamsType(inplace=bool_t)
__props__ = ('inplace',) __props__ = ('inplace',)
def __init__(self, inplace=False): def __init__(self, inplace=False):
...@@ -155,7 +155,7 @@ class GpuGemm(BlasOp): ...@@ -155,7 +155,7 @@ class GpuGemm(BlasOp):
Gemm on the GPU. Gemm on the GPU.
""" """
params_type = ParamsType(inplace=Scalar('bool')) params_type = ParamsType(inplace=bool_t)
__props__ = ('inplace',) __props__ = ('inplace',)
_f16_ok = True _f16_ok = True
...@@ -244,7 +244,7 @@ class GpuGer(BlasOp): ...@@ -244,7 +244,7 @@ class GpuGer(BlasOp):
Ger on the GPU. Ger on the GPU.
""" """
params_type = ParamsType(inplace=Scalar('bool')) params_type = ParamsType(inplace=bool_t)
__props__ = ('inplace',) __props__ = ('inplace',)
def __init__(self, inplace=False): def __init__(self, inplace=False):
...@@ -378,7 +378,7 @@ gpu_dot22 = GpuDot22() ...@@ -378,7 +378,7 @@ gpu_dot22 = GpuDot22()
class GpuGemmBatch(BlasOp): class GpuGemmBatch(BlasOp):
params_type = ParamsType(inplace=Scalar('bool')) params_type = ParamsType(inplace=bool_t)
__props__ = ('inplace',) __props__ = ('inplace',)
_f16_ok = True _f16_ok = True
......
Markdown 格式
0%
您添加了 0 到此讨论。请谨慎行事。
请先完成此评论的编辑!
注册 或者 后发表评论