Searched refs:tmp_sum_scalar (Results 1 – 1 of 1) sorted by relevance
57 auto tmp_sum_scalar = std::make_unique<scalar_t[]>(CHUNK_SIZE); in _vec_log_softmax_lastdim() local75 tmp_sum_scalar[j] = vec::map_reduce_all<scalar_t>( in _vec_log_softmax_lastdim()85 tmp_sum_scalar.get(), in _vec_log_softmax_lastdim()86 tmp_sum_scalar.get(), in _vec_log_softmax_lastdim()92 scalar_t tmp_sum = tmp_sum_scalar[j]; in _vec_log_softmax_lastdim()