Searched refs:ConvBackend (Results 1 – 5 of 5) sorted by relevance
| /external/pytorch/aten/src/ATen/native/ |
| D | Convolution.cpp | 796 const ConvBackend backend) { in check_input_same_type_as_parameters()
797 if (backend == ConvBackend::Mkldnn || backend == ConvBackend::MkldnnTranspose) { in check_input_same_type_as_parameters()
1202 ConvBackend _select_conv_backend( in _select_conv_backend()
1212 return input.is_mkldnn() ? ConvBackend::MkldnnEmpty : ConvBackend::Empty; in _select_conv_backend()
1219 return ConvBackend::Cudnn; in _select_conv_backend()
1221 return ConvBackend::MiopenDepthwise; in _select_conv_backend()
1224 return ConvBackend::CudaDepthwise2d; in _select_conv_backend()
1226 return ConvBackend::CudaDepthwise3d; in _select_conv_backend()
1233 return ConvBackend::CudnnTranspose; in _select_conv_backend()
1235 return ConvBackend::Cudnn; in _select_conv_backend()
[all …]
|
| D | ConvUtils.h | 96 enum class ConvBackend { enum
124 TORCH_API ConvBackend select_conv_backend(
131 const ConvBackend backend);
|
| /external/pytorch/torch/csrc/ |
| D | Module.cpp | 1884 py::enum_<at::native::ConvBackend>(py_module, "_ConvBackend") in initModule()
1885 .value("CudaDepthwise2d", at::native::ConvBackend::CudaDepthwise2d) in initModule()
1886 .value("CudaDepthwise3d", at::native::ConvBackend::CudaDepthwise3d) in initModule()
1887 .value("Cudnn", at::native::ConvBackend::Cudnn) in initModule()
1888 .value("CudnnTranspose", at::native::ConvBackend::CudnnTranspose) in initModule()
1889 .value("Empty", at::native::ConvBackend::Empty) in initModule()
1890 .value("Miopen", at::native::ConvBackend::Miopen) in initModule()
1891 .value("MiopenDepthwise", at::native::ConvBackend::MiopenDepthwise) in initModule()
1892 .value("MiopenTranspose", at::native::ConvBackend::MiopenTranspose) in initModule()
1893 .value("Mkldnn", at::native::ConvBackend::Mkldnn) in initModule()
[all …]
|
| /external/pytorch/docs/source/notes/ |
| D | numerical_accuracy.rst | 163 * ConvBackend::Miopen
164 * ConvBackend::MiopenDepthwise
165 * ConvBackend::MiopenTranspose
|
| /external/pytorch/torch/_C/ |
| D | __init__.pyi.in | 1190 def _select_conv_backend(*args, **kwargs) -> ConvBackend: ...
1194 backend: ConvBackend,
1264 class ConvBackend(Enum): ...
|