Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
cuda.rst 502 Bytes
Newer Older
yuguo's avatar
Oneflow 0.8 for DCU  
yuguo committed
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22 
oneflow.cuda
===================================
ONEFLOW.CUDA 
----------------------------------
.. currentmodule:: oneflow.cuda
.. automodule:: oneflow.cuda
    :members: is_available,
        device_count,
        current_device,
        set_device,
        synchronize,
        manual_seed_all,
        manual_seed,
        empty_cache,
        HalfTensor,
        FloatTensor,
        DoubleTensor,
        BoolTensor,
        ByteTensor,
        CharTensor,
        IntTensor,
        LongTensor,