Searched refs:softmax_matrix (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/sparse/ |
| H A D | softmax_op.cc | 121 const CSRSparseMatrix* softmax_matrix; in Compute() local
122 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 0, &softmax_matrix)); in Compute()
123 OP_REQUIRES(ctx, softmax_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
126 DataTypeString(softmax_matrix->dtype()), " vs. ", in Compute()
138 ctx, softmax_matrix->dims() == grad_softmax_matrix->dims(), in Compute()
141 softmax_matrix->dims(), " vs. ", grad_softmax_matrix->dims())); in Compute()
144 ctx, softmax_matrix->dims() == grad_softmax_matrix->dims(), in Compute()
147 softmax_matrix->dims(), " vs. ", grad_softmax_matrix->dims())); in Compute()
149 Tensor dense_shape_t = softmax_matrix->dense_shape(); in Compute()
167 const int total_nnz = softmax_matrix->total_nnz(); in Compute()
[all …]
|