Searched defs:ind_vec (Results 1 – 3 of 3) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
H A D | AdaptiveMaxPoolKernel.cpp | 168 iVec ind_vec = iVec::blendv(maxindex_vec, index_vec, imask); in cpu_adaptive_max_pool2d_channels_last() local 649 iVec ind_vec = iVec::blendv(maxindex_vec, index_vec, imask); in cpu_adaptive_max_pool3d_channels_last() local
|
H A D | MaxPoolKernel.cpp | 119 iVec ind_vec = iVec::blendv(maxindex_vec, index_vec, imask); in compute_internal() local
|
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/draw/ |
H A D | draw_llvm.c | 1566 LLVMValueRef ind_vec, start_vec, have_elts, fetch_max, tmp; in draw_llvm_generate() local
|