Searched defs:QuantizedCUDA (Results 1 – 8 of 8) sorted by relevance
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cudnn/ |
| H A D | LinearPrepack.cpp | 50 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | ConvPrepack.cpp | 206 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | Pooling.cpp | 239 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | BinaryOps.cpp | 251 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | Conv.cpp | 391 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| H A D | Linear.cpp | 359 TORCH_LIBRARY_IMPL(quantized, QuantizedCUDA, m) { in TORCH_LIBRARY_IMPL() argument
|
| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 55 QuantizedCUDA, enumerator
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 151 QuantizedCUDA = auto() variable in DispatchKey
|