Home
last modified time | relevance | path

Searched refs:swap_cache_idx_shape (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore/ops/operations/
D_embedding_cache_ops.py170 def check_shape(self, cache_table_shape, swap_cache_idx_shape, miss_value_shape): argument
/third_party/mindspore/mindspore/core/abstract/
Dprim_arrays.cc480 auto swap_cache_idx_shape = swap_cache_idx_shp->shape(); in InferImplCacheSwapTable() local
482 shape.emplace_back(swap_cache_idx_shape[0]); in InferImplCacheSwapTable()