Python _array_to_cpuの例

コード例 #1

ファイル: cuda.py プロジェクト: 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)

コード例 #2

ファイル: cuda.py プロジェクト: 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)