Home
last modified time | relevance | path

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

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DSoftMaxKernel.cpp1047 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local
1074 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local
1101 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local