Searched refs:cache_table (Results 1 – 3 of 3) sorted by relevance
| /third_party/mindspore/mindspore/ccsrc/frontend/parallel/cache_embedding/ |
| D | cache_embedding.cc | 373 AnfNodePtr CreateCacheSwapTable(const FuncGraphPtr &graph, ParameterPtr cache_table, AnfNodePtr swa… 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 | 471 auto cache_table = CheckArg<AbstractTensor>(op_name, args_spec_list, 0); in InferImplCacheSwapTable() local
472 auto cache_table_shp = cache_table->shape(); in InferImplCacheSwapTable()
498 …std::make_shared<AbstractTensor>(cache_table->element(), std::make_shared<Shape>(shape, min_shape,… in InferImplCacheSwapTable()
|
| /third_party/mindspore/config/ |
| D | op_info.config | 47 {"op_name": "CacheSwapTable", "inputs": [{"index": 0, "name": "cache_table", "param_type": "require…
|