Home
last modified time | relevance | path

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

/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/embedding_cache/
Ddevice_dense_embedding_operation.cc308 std::vector<abstract::AbstractBasePtr> input_kernel_tensors_for_infer(kCacheOpInputNum, nullptr); in LookupDeviceCache() local
369 std::vector<abstract::AbstractBasePtr> input_kernel_tensors_for_infer(kCacheOpInputNum, nullptr); in UpdateDeviceCache() local
Ddevice_embedding_operation.cc293 const std::vector<abstract::AbstractBasePtr> &input_kernel_tensors_for_infer) { in InferOpShape()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/graph_scheduler/actor/
Dabstract_actor.cc121 std::vector<abstract::AbstractBasePtr> *const input_kernel_tensors_for_infer, in FetchInputByTensorStore()
/third_party/mindspore/mindspore-src/source/mindspore/ccsrc/runtime/pynative/
Dop_runner.cc600 auto input_kernel_tensors_for_infer = GetInputKernelTensorsForInfer(runtime_info, node); in LaunchKernels() local