Searched refs:b_indices_mat (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_sparse_binary_op_shared.cc | 65 typename TTypes<int64>::ConstMatrix b_indices_mat, in UnionSparseIndicesAndValues() argument
76 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j, in UnionSparseIndicesAndValues()
185 const auto b_indices_mat = b_indices_t->matrix<int64>(); in Compute() local
188 UnionSparseIndicesAndValues(a_indices_mat, a_values, a_nnz, b_indices_mat, in Compute()
206 from_a ? a_indices_mat.chip<0>(idx) : b_indices_mat.chip<0>(idx); in Compute()
|
| D | sparse_add_op.cc | 98 auto b_indices_mat = b_indices->matrix<int64>(); in Compute() local
109 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j, in Compute()
160 from_a ? a_indices_mat.chip<0>(idx) : b_indices_mat.chip<0>(idx); in Compute()
|
| D | sparse_add_grad_op.cc | 73 const auto b_indices_mat = b_indices->matrix<int64>(); in Compute() local
|