Searched refs:swap_cache_idx (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 373 …reateCacheSwapTable(const FuncGraphPtr &graph, ParameterPtr cache_table, AnfNodePtr swap_cache_idx, in CreateCacheSwapTable() argument
377 …ePtr> cache_swap_table_nodes{NewValueNode(cache_swap_table_primitive), cache_table, swap_cache_idx, in CreateCacheSwapTable()
|
| /third_party/mindspore/mindspore/core/abstract/ |
| D | prim_arrays.cc | 475 auto swap_cache_idx = CheckArg<AbstractTensor>(op_name, args_spec_list, 1); in InferImplCacheSwapTable() local
476 auto swap_cache_idx_shp = swap_cache_idx->shape(); in InferImplCacheSwapTable()
|
| /third_party/mindspore/config/ |
| D | op_info.config | 47 …me": "cache_table", "param_type": "required"}, {"index": 1, "name": "swap_cache_idx", "param_type"…
|