Searched defs:BuildEmbeddingCacheUpdateKernel (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/ | ||
| D | device_sparse_embedding_operation.cc | 178 void DeviceSparseEmbeddingOperation::BuildEmbeddingCacheUpdateKernel() { in BuildEmbeddingCacheUpdateKernel() function in mindspore::runtime::DeviceSparseEmbeddingOperation |
| D | device_dense_embedding_operation.cc | 252 void DeviceDenseEmbeddingOperation::BuildEmbeddingCacheUpdateKernel() { in BuildEmbeddingCacheUpdateKernel() function in mindspore::runtime::DeviceDenseEmbeddingOperation |