mxnet
|
This is the complete list of members for OpResource, including all inherited members.
alloc_cpu(int size) const | OpResource | inline |
alloc_gpu(int size) const | OpResource | inline |
alloc_sparse(MXSparse *sparse, int index, int indices_len, int indptr_len=0) const | OpResource | inline |
get_cpu_rand_states() const | OpResource | inline |
get_cuda_stream() const | OpResource | inline |
get_gpu_rand_states() const | OpResource | inline |
OpResource(xpu_malloc_t cpu_malloc_fp, void *cpu_alloc_fp, xpu_malloc_t gpu_malloc_fp, void *gpu_alloc_fp, void *stream, sparse_malloc_t sparse_malloc_fp, void *sparse_alloc_fp, void *rng_cpu_states, void *rng_gpu_states) | OpResource | inline |