Searched defs:tmp_sum_data (Results 1 – 1 of 1) sorted by relevance
284 scalar_t* tmp_sum_data = buffer.get(); in _vec_softmax_backward() local369 float* tmp_sum_data = buffer.get(); in _vec_softmax_backward() local495 scalar_t* tmp_sum_data = buffer.get(); in _vec_log_softmax_backward() local579 float* tmp_sum_data = buffer.get(); in _vec_log_softmax_backward() local911 scalar_t* tmp_sum_data = buffer.get() + CHUNK_SIZE; in _vec_logsoftmax() local1016 float* tmp_sum_data = buffer.get() + CHUNK_SIZE; in _vec_logsoftmax() local