Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/cuda/detail/
DCUDAHooks.h38 bool compiledWithMIOpen() const override;
DCUDAHooks.cpp264 bool CUDAHooks::compiledWithMIOpen() const { in compiledWithMIOpen() function in at::cuda::detail::CUDAHooks
/external/pytorch/aten/src/ATen/detail/
DCUDAHooksInterface.h136 virtual bool compiledWithMIOpen() const { in compiledWithMIOpen() function
/external/pytorch/aten/src/ATen/native/
DConvUtils.h359 if (!at::detail::getCUDAHooks().compiledWithMIOpen() || in miopen_conv_use_channels_last()
DConvolution.cpp511 && detail::getCUDAHooks().compiledWithMIOpen() in use_miopen()
1430 … if (detail::getCUDAHooks().compiledWithMIOpen() && miopen_conv_use_channels_last(input, weight)) { in determine_backend_memory_format()
DNormalization.cpp523 && detail::getCUDAHooks().compiledWithMIOpen() in _select_batch_norm_backend()
DRNN.cpp74 (detail::getCUDAHooks().compiledWithMIOpen()) && in use_miopen()