Skip to content

GitLab

"cacheflow/model_executor/layers/sampler.py" did not exist on "de0fabbc5c84e6771d70b92014ae06fe82654ff0"
Switch branch/tag
Find file Normal viewHistoryPermalink
tensor_attributes.rst 786 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 
.. currentmodule:: oneflow

Tensor Attributes
=============================================================
Each local ``oneflow.Tensor`` has a :class:`oneflow.dtype`, :class:`oneflow.device`, and global ``oneflow.Tensor`` has a :class:`oneflow.dtype`, :class:`oneflow.placement`, :class:`oneflow.sbp`.

oneflow.device
--------------------------------------------------------------
.. autoclass:: oneflow.device

oneflow.placement
--------------------------------------------------------------
.. autoclass:: oneflow.placement

oneflow.env.all_device_placement
--------------------------------------------------------------
.. autofunction:: oneflow.env.all_device_placement

oneflow.sbp.sbp
--------------------------------------------------------------
.. autoclass:: oneflow.sbp.sbp