Home
last modified time | relevance | path

Searched refs:AT_DISPATCH_SWITCH (Results 1 – 16 of 16) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DDispatch.h213 #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 DSparseCsrTensorUtils.h134 AT_DISPATCH_SWITCH( \
H A DDispatch_v2.h81 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 DMixedDtypesLinear.cu289 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 DUnaryComplexKernels.cu85 AT_DISPATCH_SWITCH(iter.common_dtype(), "conj_cuda", in conj_kernel_cuda()
H A DForeachBinaryOpList.cu289 AT_DISPATCH_SWITCH( \
H A DForeachReduceOp.cu390 AT_DISPATCH_SWITCH( \
H A DLoss.cu160 AT_DISPATCH_SWITCH(TYPE, NAME, \
/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/
H A DCUDASymmetricMemoryOps.cu152 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 DTensorFactory.h4 AT_DISPATCH_SWITCH( \
/aosp_15_r20/external/pytorch/aten/src/ATen/native/ao_sparse/quantized/cpu/
H A Dqlinear_deserialize.cpp294 AT_DISPATCH_SWITCH( \ in PackedLinearWeightQnnp()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DUnaryOpsKernel.cpp220 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 DFlashAttentionKernel.cpp715 AT_DISPATCH_SWITCH( \
/aosp_15_r20/external/pytorch/aten/src/ATen/native/sparse/cuda/
H A DSparseSemiStructuredLinear.cu707 AT_DISPATCH_SWITCH( in _sparse_semi_structured_linear()
H A DSparseSemiStructuredOps.cu619 AT_DISPATCH_SWITCH( in sparse_semi_structured_mad_op()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu1504 AT_DISPATCH_SWITCH( in _fbgemm_dense_to_jagged_forward_symint()