Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp701 …int vectorized_step = Vec16().size(); // Currently, we only support BFloat16/Half in this special … in _vec_softmax() local
808 int vectorized_step = Vec().size(); in _vec_softmax() local