Searched refs:DimComparator (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | dim_comparator.h | 45 class DimComparator {
49 DimComparator(const TTypes<int64>::Matrix& ix, const VarDimArray& order, in DimComparator() function
95 class FixedDimComparator : DimComparator {
99 : DimComparator(ix, order, shape) { in FixedDimComparator()
|
| D | sparse_tensor_test.cc | 71 DimComparator sorter(map, order, shape); in TEST()
81 DimComparator sorter1(map, order1, shape); in TEST()
|
| D | sparse_tensor.h | 406 DimComparator sorter(ix_t, order, shape()); in Reorder()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | sparse_add_grad_op.cc | 89 switch (sparse::DimComparator::cmp(a_or_b##_indices_mat, sum_indices_mat, \ in Compute()
|
| D | sparse_add_op.cc | 109 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j, in Compute()
|
| D | sparse_sparse_binary_op_shared.cc | 76 switch (sparse::DimComparator::cmp(a_indices_mat, b_indices_mat, i, j, in UnionSparseIndicesAndValues()
|