Searched refs:input_buffer_ptr (Results 1 – 1 of 1) sorted by relevance
1047 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local1061 data_fvec0.store(input_buffer_ptr + d1); in _vec_logsoftmax()1062 data_fvec1.store(input_buffer_ptr + d1 + fVec::size()); in _vec_logsoftmax()1068 input_buffer_ptr[d1] = data_val; in _vec_logsoftmax()1074 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local1078 fVec data_fvec0 = fVec::loadu(input_buffer_ptr + d2); in _vec_logsoftmax()1079 fVec data_fvec1 = fVec::loadu(input_buffer_ptr + d2 + fVec::size()); in _vec_logsoftmax()1090 float data_val = input_buffer_ptr[d2]; in _vec_logsoftmax()1101 float* input_buffer_ptr = input_buffer_data + dim_idx * CHUNK_SIZE; in _vec_logsoftmax() local1107 fVec data_fvec0 = fVec::loadu(input_buffer_ptr + d3); in _vec_logsoftmax()[all …]