Home
last modified time | relevance | path

Searched defs:embedding_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ccsrc/backend/kernel_compiler/cpu/nnacl/infer/
Dembedding_lookup_infer.c41 int embedding_shape[MAX_SHAPE_SIZE] = {0}; in EmbeddingLookupInferShape() local
/third_party/mindspore/tests/st/networks/models/bert/src/
Dbert_model.py118 embedding_shape, argument
169 embedding_shape, argument