Searched defs:BATCH_SLICE (Results 1 – 2 of 2) sorted by relevance
| /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 |