Home
last modified time | relevance | path

Searched defs:params_shp (Results 1 – 4 of 4) sorted by relevance

/third_party/mindspore/mindspore/core/ops/
Dembedding_lookup.cc54 auto params_shp = params->shape()->shape(); in EmbeddingLookupInfer() local
Dgather.cc65 auto params_shp = params->shape()->shape(); in GatherInfer() local
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc605 auto params_shp = params->shape()->shape(); in InferImplGatherV2() local
667 auto params_shp = params->shape(); in InferImplEmbeddingLookup() local
/third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/
Dcache_embedding.cc391 auto params_shp = params_ori_shp->cast<abstract::ShapePtr>(); in CreateUpdateCache() local