Home
last modified time | relevance | path

Searched refs:cudaMemcpyKind (Results 1 – 3 of 3) sorted by relevance

/third_party/mindspore/tests/ut/cpp/stub/runtime/
Dcuda_runtime_api.h24 enum cudaMemcpyKind { enum
39 cudaError_t cudaMemcpy(void *dst, const void *src, size_t count, enum cudaMemcpyKind kind);
Dcuda_runtime_api.cc22 cudaError_t cudaMemcpy(void *dst, const void *src, size_t count, enum cudaMemcpyKind kind) { return… in cudaMemcpy()
/third_party/mindspore/mindspore/lite/src/delegate/tensorrt/
Dtensorrt_allocator.cc115 cudaMemcpyKind kind = is_host2device ? cudaMemcpyHostToDevice : cudaMemcpyDeviceToHost; in SyncMemInHostAndDevice()