Searched refs:AT_DISPATCH_SWITCH (Results 1 – 16 of 16) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/ |
H A D | Dispatch.h | 213 #define AT_DISPATCH_SWITCH(TYPE, NAME, ...) \ macro 237 AT_DISPATCH_SWITCH(TYPE, NAME, AT_DISPATCH_CASE_FLOATING_TYPES(__VA_ARGS__)) 245 AT_DISPATCH_SWITCH( \ 253 AT_DISPATCH_SWITCH( \ 261 AT_DISPATCH_SWITCH( \ 273 AT_DISPATCH_SWITCH( \ 288 AT_DISPATCH_SWITCH( \ 304 AT_DISPATCH_SWITCH( \ 315 AT_DISPATCH_SWITCH(TYPE, NAME, AT_DISPATCH_CASE_COMPLEX_TYPES(__VA_ARGS__)) 322 AT_DISPATCH_SWITCH( \ [all …]
|
H A D | SparseCsrTensorUtils.h | 134 AT_DISPATCH_SWITCH( \
|
H A D | Dispatch_v2.h | 81 AT_DISPATCH_SWITCH(TYPE, NAME, AT_AP_VAR(AT_WRAP(BODY), TYPE, __VA_ARGS__))
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | MixedDtypesLinear.cu | 289 AT_DISPATCH_SWITCH( in _mixed_dtypes_linear() 295 AT_DISPATCH_SWITCH( in _mixed_dtypes_linear() 322 AT_DISPATCH_SWITCH( in _mixed_dtypes_linear()
|
H A D | UnaryComplexKernels.cu | 85 AT_DISPATCH_SWITCH(iter.common_dtype(), "conj_cuda", in conj_kernel_cuda()
|
H A D | ForeachBinaryOpList.cu | 289 AT_DISPATCH_SWITCH( \
|
H A D | ForeachReduceOp.cu | 390 AT_DISPATCH_SWITCH( \
|
H A D | Loss.cu | 160 AT_DISPATCH_SWITCH(TYPE, NAME, \
|
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/ |
H A D | CUDASymmetricMemoryOps.cu | 152 AT_DISPATCH_SWITCH( in multimem_all_reduce_() 237 AT_DISPATCH_SWITCH( in multimem_one_shot_all_reduce()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/mps/ |
H A D | TensorFactory.h | 4 AT_DISPATCH_SWITCH( \
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/ |
H A D | qlinear_deserialize.cpp | 294 AT_DISPATCH_SWITCH( \ in PackedLinearWeightQnnp()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | UnaryOpsKernel.cpp | 220 AT_DISPATCH_SWITCH(iter.common_dtype(), "conj_cpu", in conj_kernel() 322 AT_DISPATCH_SWITCH(iter.input_dtype(), "signbit_cpu", in signbit_kernel()
|
H A D | FlashAttentionKernel.cpp | 715 AT_DISPATCH_SWITCH( \
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/ |
H A D | SparseSemiStructuredLinear.cu | 707 AT_DISPATCH_SWITCH( in _sparse_semi_structured_linear()
|
H A D | SparseSemiStructuredOps.cu | 619 AT_DISPATCH_SWITCH( in sparse_semi_structured_mad_op()
|
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/ |
H A D | NestedTensorTransformerFunctions.cu | 1504 AT_DISPATCH_SWITCH( in _fbgemm_dense_to_jagged_forward_symint()
|