Searched refs:b_nnz (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_sparse_binary_op_shared.cc | 66 typename TTypes<T>::ConstFlat b_values, int64 b_nnz, int num_dims, in UnionSparseIndicesAndValues() argument
69 entries_to_copy->reserve(a_nnz + b_nnz); in UnionSparseIndicesAndValues()
71 b_augmented_values->reserve(b_nnz); in UnionSparseIndicesAndValues()
75 while (i < a_nnz && j < b_nnz) { in UnionSparseIndicesAndValues()
105 while (j < b_nnz) { in UnionSparseIndicesAndValues()
152 const int64 b_nnz = b_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()
189 b_values, b_nnz, num_dims, &a_augmented_values, in Compute()
|
| D | sparse_add_grad_op.cc | 69 const int64 b_nnz = b_indices->dim_size(0); in Compute() local
80 ctx->allocate_output(1, TensorShape({b_nnz}), &b_val_grad)); in Compute()
86 memset(b_val_grad_flat, 0, sizeof(T) * b_nnz); in Compute()
107 while (i < a_nnz && j < b_nnz && k < sum_nnz) { in Compute()
122 while (j < b_nnz && k < sum_nnz) { in Compute()
|
| D | sparse_add_op.cc | 45 const int64 b_nnz = b_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()
|