Searched refs:grad_softmax_matrix (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | softmax_op.cc | 129 const CSRSparseMatrix* grad_softmax_matrix; in Compute() local
130 OP_REQUIRES_OK(ctx, ExtractVariantFromInput(ctx, 1, &grad_softmax_matrix)); in Compute()
131 OP_REQUIRES(ctx, grad_softmax_matrix->dtype() == DataTypeToEnum<T>::value, in Compute()
134 DataTypeString(grad_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()
154 grad_softmax_matrix->dense_shape().vec<int64>(); in Compute()
161 grad_softmax_matrix->dense_shape().SummarizeValue(3))); in Compute()
[all …]
|