Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc316 ShapeVector input_param_shape = {SizeToLong(cache_size), SizeToLong(embedding_size)}; in LookupDeviceCache() local
377 ShapeVector input_param_shape = {SizeToLong(cache_size), SizeToLong(embedding_size)}; in UpdateDeviceCache() local