Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/torch/csrc/distributed/c10d/quantization/
H A Dquantization_gpu.cu15 const auto row_incre = blockDim.y * gridDim.y; in _float_to_bfloat16_cuda_kernel() local
37 const auto row_incre = blockDim.y * gridDim.y; in _bfloat16_to_float_cuda_kernel() local