Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/miopen/
DHandle.cpp31 using MIOpenPoolType = at::cuda::DeviceThreadHandlePool<miopenHandle_t, createMIOpenHandle, destroy… typedef
44 static auto pool = std::make_shared<MIOpenPoolType>(); in getMiopenHandle()
45 thread_local std::unique_ptr<MIOpenPoolType::PoolWindow> myPoolWindow( in getMiopenHandle()