Python GpuArrayType.name Beispiele

Beispiel #1

Datei: opt.py Projekt: jlowin/Theano

def tensor_to_gpu(x):
    if isinstance(x.type, tensor.TensorType):
        y = GpuArrayType(broadcastable=x.type.broadcastable, dtype=x.type.dtype)()
        if x.name:
            y.name = x.name + "[Gpua]"
        return y
    else:
        return x

Beispiel #2

Datei: opt.py Projekt: kevinbache/Theano

def tensor_to_gpu(x):
    if isinstance(x.type, tensor.TensorType):
        y = GpuArrayType(broadcastable=x.type.broadcastable,
                         dtype=x.type.dtype)()
        if x.name:
            y.name = x.name + '[Gpua]'
        return y
    else:
        return x