Searched refs:ix_t (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/sparse/ |
| D | sparse_tensor_test.cc | 183 auto ix_t = ix.matrix<int64>(); in TEST() local
186 ix_t = ix_c; in TEST()
207 ix_t = ix_c; in TEST()
217 ix_t = ix_c; in TEST()
253 auto ix_t = ix.matrix<int64>(); in TEST() local
256 ix_t = ix_t.random(Eigen::internal::UniformRandomGenerator<int64>(n + 1)); in TEST()
257 ix_t = ix_t.abs() % 1000; in TEST()
285 auto ix_t = ix.matrix<int64>(); in TEST() local
286 ix_t = ix_orig; in TEST()
300 ix_t = ix_orig; in TEST()
[all …]
|
| D | group_iterator.cc | 24 const auto& ix_t = iter_->ix_matrix_; in UpdateEndOfGroup() local
25 const int64 N = ix_t.dimension(0); in UpdateEndOfGroup()
26 while (next_loc_ < N && iter_->GroupMatches(ix_t, loc_, next_loc_)) { in UpdateEndOfGroup()
57 const auto& ix_t = iter_->ix_matrix_; in group() local
59 g.push_back(ix_t(loc_, d)); in group()
|
| D | sparse_tensor.h | 167 const auto ix_t = ix_.matrix<int64>(); in IndicesValid() local
177 TF_RETURN_IF_ERROR(IndexValid(ix_t, n)); in IndicesValid()
286 inline Status IndexValid(const TTypes<int64>::ConstMatrix& ix_t, in IndexValid() argument
293 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_[di]) valid = false; in IndexValid()
298 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_[di]) valid = false; in IndexValid()
299 int64 diff = ix_t(n, order_[di]) - ix_t(n - 1, order_[di]); in IndexValid()
307 strings::StrAppend(&index, ix_t(n, di), di < dims_ - 1 ? "," : "]"); in IndexValid()
384 auto ix_t = ix_.matrix<int64>(); in Reorder() local
394 FixedDimComparator<ORDER_SIZE> sorter(ix_t, order, shape()); \ in Reorder()
406 DimComparator sorter(ix_t, order, shape()); in Reorder()
[all …]
|
| /external/tensorflow/tensorflow/core/util/ |
| D | example_proto_helper.cc | 158 auto ix_t = indices->matrix<int64>(); in CopyIntoSparseTensor() local
159 int64* ix_p = &ix_t(offset, 0); in CopyIntoSparseTensor()
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | barrier_ops_test.py | 266 for ix_t, k_t, v_t in take_t:
267 index_fetches.append(ix_t)
|