Home
last modified time | relevance | path

Searched defs:BuildEmbeddingCacheLookupKernel (Results 1 – 2 of 2) sorted by relevance

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_sparse_embedding_operation.cc149 void DeviceSparseEmbeddingOperation::BuildEmbeddingCacheLookupKernel() { in BuildEmbeddingCacheLookupKernel() function in mindspore::runtime::DeviceSparseEmbeddingOperation
Ddevice_dense_embedding_operation.cc218 void DeviceDenseEmbeddingOperation::BuildEmbeddingCacheLookupKernel() { in BuildEmbeddingCacheLookupKernel() function in mindspore::runtime::DeviceDenseEmbeddingOperation