Home
last modified time | relevance | path

Searched defs:XQ (Results 1 – 1 of 1) sorted by relevance

/external/pytorch/aten/src/ATen/native/cuda/
DRowwiseScaledMM.cu139 at::Tensor XQ, // FP8 in f8f8bf16_rowwise_impl()
307 at::Tensor XQ, in dispatch_fp8_rowwise_kernel_on_tile_size()
344 at::Tensor XQ, in handle_transposition()
371 at::Tensor XQ, in dispatch_fp8_rowwise_kernel_on_cluster_size_and_transpose()
456 at::Tensor XQ, in dispatch_fp8_rowwise_kernel_on_fast_accum()
476 at::Tensor XQ, in dispatch_fp8_rowwise_kernel_on_input_dtypes()
497 at::Tensor XQ, in dispatch_fp8_rowwise_kernel_on_bias_dtype()
570 at::Tensor XQ, // FP8 in f8f8bf16_rowwise()