Searched refs:logits_values (Results 1 – 1 of 1) sorted by relevance
438 auto logits_values = logits.values().vec<T>(); in CSRSparseMatrixSoftmaxGPUImpl() local450 logits_values.data(), softmax_values.data())); in CSRSparseMatrixSoftmaxGPUImpl()472 logits_values.data(), softmax_values.data())); in CSRSparseMatrixSoftmaxGPUImpl()