Searched defs:GetKernelTensor (Results 1 – 2 of 2) sorted by relevance
| /third_party/mindspore/mindspore-src/source/mindspore/ccsrc/kernel/pyboost/ |
| D | pyboost_utils.h | 116 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, size_t index, in GetKernelTensor() function
128 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor() function
158 static void GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor() function
|
| D | pyboost_utils.cc | 292 void PyBoostUtils::GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor() function in mindspore::kernel::pyboost::PyBoostUtils
308 void PyBoostUtils::GetKernelTensor(const DeviceContext *device_context, size_t stream_id, in GetKernelTensor() function in mindspore::kernel::pyboost::PyBoostUtils
|