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