Searched defs:softmax_values (Results 1 – 1 of 1) sorted by relevance
434 typename TTypes<T>::Vec softmax_values) { in CSRSparseMatrixSoftmaxGPUImpl()628 auto softmax_values = softmax.values().vec<T>(); in CSRSparseMatrixSoftmaxGradGPUImpl() local