Home
last modified time | relevance | path

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

/external/pytorch/aten/src/ATen/native/cpu/
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