Searched refs:MIOpenAllocatorCallback (Results 1 – 1 of 1) sorted by relevance
2572 void* MIOpenAllocatorCallback(void* ctx, size_t size_in_bytes) { in MIOpenAllocatorCallback() function2625 wrap::miopenSetAllocator(miopen.handle(), MIOpenAllocatorCallback, in DoPrepareForConvolution()