Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization_gpu.cu10 __global__ void _float_to_bfloat16_cuda_kernel( in _float_to_bfloat16_cuda_kernel() function
87 _float_to_bfloat16_cuda_kernel<<< in _float_to_bfloat16_cuda()