Searched refs:a_nnz (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_sparse_binary_op_shared.cc | 64 typename TTypes<T>::ConstFlat a_values, int64 a_nnz, in UnionSparseIndicesAndValues() argument
69 entries_to_copy->reserve(a_nnz + b_nnz); in UnionSparseIndicesAndValues()
70 a_augmented_values->reserve(a_nnz); in UnionSparseIndicesAndValues()
75 while (i < a_nnz && j < b_nnz) { in UnionSparseIndicesAndValues()
100 while (i < a_nnz) { in UnionSparseIndicesAndValues()
151 const int64 a_nnz = a_indices_t->dim_size(0); in Compute() local
157 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz, in Compute()
158 errors::InvalidArgument("Expected ", a_nnz, " and ", b_nnz, in Compute()
188 UnionSparseIndicesAndValues(a_indices_mat, a_values, a_nnz, b_indices_mat, in Compute()
|
| D | sparse_add_grad_op.cc | 68 const int64 a_nnz = a_indices->dim_size(0); in Compute() local
78 ctx->allocate_output(0, TensorShape({a_nnz}), &a_val_grad)); in Compute()
85 memset(a_val_grad_flat, 0, sizeof(T) * a_nnz); in Compute()
107 while (i < a_nnz && j < b_nnz && k < sum_nnz) { in Compute()
117 while (i < a_nnz && k < sum_nnz) { in Compute()
|
| D | sparse_add_op.cc | 44 const int64 a_nnz = a_indices->dim_size(0); in Compute() local
60 ctx, a_values.size() == a_nnz && b_values.size() == b_nnz, in Compute()
61 errors::InvalidArgument("Expected ", a_nnz, " and ", b_nnz, in Compute()
100 entries_to_copy.reserve(a_nnz + b_nnz); in Compute()
108 while (i < a_nnz && j < b_nnz) { in Compute()
|