| /aosp_15_r20/external/pytorch/c10/core/ |
| H A D | Backend.h | 55 QuantizedCUDA, enumerator
117 } else if (t == DispatchKey::QuantizedCUDA) { in dispatchKeyToBackend()
118 return Backend::QuantizedCUDA; in dispatchKeyToBackend()
201 case Backend::QuantizedCUDA: in backendToDispatchKey()
202 return DispatchKey::QuantizedCUDA; in backendToDispatchKey()
230 case Backend::QuantizedCUDA: in backendToDeviceType()
342 case Backend::QuantizedCUDA: in toString()
343 return "QuantizedCUDA"; in toString()
|
| H A D | DispatchKey.cpp | 338 {"QuantizedCUDA", c10::DispatchKey::QuantizedCUDA}, in parseDispatchKey()
|
| H A D | DispatchKeySet.h | 83 // - "Quantized": QuantizedCPU, QuantizedCUDA, QuantizedXLA, ...
|
| /aosp_15_r20/external/pytorch/c10/test/core/ |
| H A D | DispatchKeySet_test.cpp | 337 // quantizedCUDA has higher priority than CUDA in TEST()
339 {DispatchKey::CUDA, DispatchKey::QuantizedCUDA}); in TEST()
341 DispatchKey::QuantizedCUDA, in TEST()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | native_functions.yaml | 937 QuantizedCPU, QuantizedCUDA: as_strided_qtensorimpl
2375 QuantizedCPU, QuantizedCUDA, QuantizedMeta: empty_unknown_quantized
2425 QuantizedCPU, QuantizedCUDA: empty_affine_quantized
2434 QuantizedCPU, QuantizedCUDA: empty_per_channel_affine_quantized
2466 QuantizedCPU, QuantizedCUDA: empty_quantized
2478 QuantizedCPU, QuantizedCUDA: empty_like_quantized
2490 QuantizedCPU, QuantizedCUDA: empty_strided_unknown_quantized
2684 QuantizedCPU, QuantizedCUDA: fill_quantized_
2696 QuantizedCPU, QuantizedCUDA: fill_quantized_
3117 QuantizedCUDA: _index_put_impl_quantized_cuda_
[all …]
|
| /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 | Linear.cpp | 359 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
|
| /aosp_15_r20/external/pytorch/torchgen/ |
| H A D | model.py | 151 QuantizedCUDA = auto() variable in DispatchKey
285 DispatchKey.QuantizedCUDA,
318 DispatchKey.QuantizedCUDA,
|
| /aosp_15_r20/external/pytorch/torchgen/dest/ |
| H A D | register_dispatch_key.py | 101 DispatchKey.QuantizedCUDA,
|
| /aosp_15_r20/external/pytorch/test/quantization/fx/ |
| H A D | test_quantize_fx.py | 9723 # which doesn't support QuantizedCuda backend
|