| /external/tensorflow/tensorflow/core/kernels/ |
| sparse_sparse_binary_op_shared.cc | 63 typename TTypes<int64>::ConstMatrix a_indices_mat,
76 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j,
184 const auto a_indices_mat = a_indices_t->matrix<int64>(); variable
188 UnionSparseIndicesAndValues(a_indices_mat, a_values, a_nnz, b_indices_mat,
206 from_a ? a_indices_mat.chip<0>(idx) : b_indices_mat.chip<0>(idx);
|
| sparse_tensor_dense_add_op.cc | 87 const auto a_indices_mat = a_indices_t->flat_inner_dims<Index>(); variable
98 ctx->eigen_device<Device>(), a_indices_mat, a_values_flat, \
|
| sparse_add_op.cc | 97 auto a_indices_mat = a_indices->matrix<int64>(); variable
109 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j,
160 from_a ? a_indices_mat.chip<0>(idx) : b_indices_mat.chip<0>(idx);
|
| sparse_add_grad_op.cc | 72 const auto a_indices_mat = a_indices->matrix<int64>(); variable
|
