Home
last modified time | relevance | path

Searched refs:TABLE_ROW_SLICE (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/python/parallel/
Dtest_multi_field_embedding.py95 …net = NetWithLoss(Net(shape, field_size=9, slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, target='…
123 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, opera…
144 …net = NetWithLoss(Net(shape, target='DEVICE', slice_mode=nn.EmbeddingLookup.TABLE_ROW_SLICE, opera…
/third_party/mindspore/mindspore/nn/layer/
Dembedding.py220 TABLE_ROW_SLICE = "table_row_slice" variable in EmbeddingLookup
Dthor_layer.py717 TABLE_ROW_SLICE = "table_row_slice" variable in EmbeddingLookupThor