mxnet
Functions
dnnl_ocl.h File Reference
#include "oneapi/dnnl/dnnl.h"
#include "oneapi/dnnl/dnnl_ocl_types.h"
Include dependency graph for dnnl_ocl.h:
This graph shows which files directly or indirectly include this file:

Go to the source code of this file.

Functions

dnnl_status_t DNNL_API dnnl_ocl_interop_memory_create (dnnl_memory_t *memory, const dnnl_memory_desc_t *memory_desc, dnnl_engine_t engine, dnnl_ocl_interop_memory_kind_t memory_kind, void *handle)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_memory_get_memory_kind (const_dnnl_memory_t memory, dnnl_ocl_interop_memory_kind_t *memory_kind)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_memory_get_mem_object (const_dnnl_memory_t memory, cl_mem *mem_object)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_memory_set_mem_object (dnnl_memory_t memory, cl_mem mem_object)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_engine_create (dnnl_engine_t *engine, cl_device_id device, cl_context context)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_engine_get_context (dnnl_engine_t engine, cl_context *context)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_get_device (dnnl_engine_t engine, cl_device_id *device)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_stream_create (dnnl_stream_t *stream, dnnl_engine_t engine, cl_command_queue queue)
 
dnnl_status_t DNNL_API dnnl_ocl_interop_stream_get_command_queue (dnnl_stream_t stream, cl_command_queue *queue)