Searched refs:a_indices_mat (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_sparse_binary_op_shared.cc | 63 typename TTypes<int64>::ConstMatrix a_indices_mat, in UnionSparseIndicesAndValues() argument
76 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j, in UnionSparseIndicesAndValues()
184 const auto a_indices_mat = a_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 | 97 auto a_indices_mat = a_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_tensor_dense_add_op.cc | 87 const auto a_indices_mat = a_indices_t->flat_inner_dims<Index>(); in Compute() local
98 ctx->eigen_device<Device>(), a_indices_mat, a_values_flat, \ in Compute()
|
| D | sparse_add_grad_op.cc | 72 const auto a_indices_mat = a_indices->matrix<int64>(); in Compute() local
|