BATCH_SLICE (reference) in projects: third_party

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

/third_party/mindspore/tests/ut/python/parallel/
D	test_multi_field_embedding.py	`59 …def __init__(self, shape, field_size=10, slice_mode=nn.EmbeddingLookup.BATCH_SLICE, target="Device…`
/third_party/mindspore/mindspore/nn/layer/
D	embedding.py	`218 BATCH_SLICE = "batch_slice" variable in EmbeddingLookup`
D	thor_layer.py	`715 BATCH_SLICE = "batch_slice" variable in EmbeddingLookupThor`