Exemplos de _array_to_cpu em Python

Exemplo n.º 1

Arquivo: cuda.py Projeto: zghzdxs/chainer

def _array_to_cpu(array, stream):
    if array is None:
        return None
    if isinstance(array, ndarray):
        check_cuda_available()
        with get_device_from_array(array):
            return array.get(stream)
    return _cpu._array_to_cpu(array)

Exemplo n.º 2

Arquivo: cuda.py Projeto: jnishi/chainer

def _array_to_cpu(array, stream):
    if array is None:
        return None
    if isinstance(array, ndarray):
        check_cuda_available()
        with get_device_from_array(array):
            return array.get(stream)
    return _cpu._array_to_cpu(array)