Searched refs:DISPATCH_TO_CUDA (Results 1 – 1 of 1) sorted by relevance
153 #define DISPATCH_TO_CUDA(name, function) \ macro157 DISPATCH_TO_CUDA("_Bfloat16QuantizedToFloat", _bfloat16_to_float_cuda); in TORCH_LIBRARY_IMPL()158 DISPATCH_TO_CUDA("_FloatToBfloat16Quantized", _float_to_bfloat16_cuda); in TORCH_LIBRARY_IMPL()