Searched refs:softmax_values (Results 1 – 3 of 3) sorted by relevance
434 typename TTypes<T>::Vec softmax_values) { in CSRSparseMatrixSoftmaxGPUImpl() argument450 logits_values.data(), softmax_values.data())); in CSRSparseMatrixSoftmaxGPUImpl()472 logits_values.data(), softmax_values.data())); in CSRSparseMatrixSoftmaxGPUImpl()482 typename TTypes<T>::Vec softmax_values) { \483 return CSRSparseMatrixSoftmaxGPUImpl<T>(ctx, logits, softmax_values); \628 auto softmax_values = softmax.values().vec<T>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local645 softmax_row_ptr.data(), softmax_col_ind.data(), softmax_values.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()677 softmax_col_ind.data(), softmax_values.data(), in CSRSparseMatrixSoftmaxGradGPUImpl()
215 typename TTypes<T>::Vec softmax_values);
104 typename TTypes<T>::Vec softmax_values); \