Skip to content

GitLab

Switch branch/tag
Find file Normal viewHistoryPermalink
one_embedding.rst 1.19 KB
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
23
24
25
26 
oneflow.one_embedding
===================================
OneFlow one_embedding operations.
----------------------------------
.. currentmodule:: oneflow.one_embedding
.. autoclass:: MultiTableEmbedding
    :members: forward,
              save_snapshot,
              load_snapshot,
.. autofunction:: oneflow.one_embedding.MultiTableEmbedding.forward
.. autoclass:: MultiTableMultiColumnEmbedding
    :members: forward,
              save_snapshot,
              load_snapshot,
.. autofunction:: oneflow.one_embedding.MultiTableMultiColumnEmbedding.forward
.. autofunction:: oneflow.one_embedding.make_device_mem_store_options
.. autofunction:: oneflow.one_embedding.make_cached_ssd_store_options       
.. autofunction:: oneflow.one_embedding.make_cached_host_mem_store_options
.. autofunction:: oneflow.one_embedding.make_uniform_initializer
.. autofunction:: oneflow.one_embedding.make_normal_initializer
.. autofunction:: oneflow.one_embedding.make_table_options
.. autofunction:: oneflow.one_embedding.make_table
.. automodule:: oneflow.one_embedding
    :members: Ftrl
.. autofunction:: oneflow.one_embedding.make_persistent_table_reader
.. autofunction:: oneflow.one_embedding.make_persistent_table_writer