Searched defs:amax_ptr (Results 1 – 2 of 2) sorted by relevance
| /external/pytorch/aten/src/ATen/cuda/tunable/ | ||
| D | GemmCommon.h | 325 void* amax_ptr; member |
| /external/pytorch/aten/src/ATen/cuda/ | ||
| D | CUDABlas.cpp | 1411 void* amax_ptr, in scaled_gemm() |