Home
last modified time | relevance | path

Searched refs:ix_t (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/core/util/sparse/
Dsparse_tensor.cc133 const auto ix_t = ix_.matrix<int64>(); in IndicesValidVectorFastPath() local
134 const int64* const index_base_ptr = ix_t.data(); in IndicesValidVectorFastPath()
136 for (std::size_t n = 0; n < ix_t.dimension(0); ++n) { in IndicesValidVectorFastPath()
155 const auto ix_t = ix_.matrix<int64>(); in IndicesValidMatrix32BitFastPath() local
182 reinterpret_cast<const int32*>(ix_t.data()); in IndicesValidMatrix32BitFastPath()
185 for (std::size_t n = 0; n < ix_t.dimension(0); ++n) { in IndicesValidMatrix32BitFastPath()
226 const auto ix_t = ix_.matrix<int64>(); in IndicesValidHelper() local
235 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_ptr[di]) valid = false; in IndicesValidHelper()
240 if (ix_t(n, di) < 0 || ix_t(n, di) >= shape_ptr[di]) valid = false; in IndicesValidHelper()
247 int64 diff = ix_t(n, ordered_dim) - ix_t(n - 1, ordered_dim); in IndicesValidHelper()
[all …]
Dsparse_tensor_test.cc183 auto ix_t = ix.matrix<int64>(); in TEST() local
186 ix_t = ix_c; in TEST()
211 ix_t = ix_c; in TEST()
221 ix_t = ix_c; in TEST()
257 auto ix_t = ix.matrix<int64>(); in TEST() local
260 ix_t = ix_t.random(Eigen::internal::UniformRandomGenerator<int64>(n + 1)); in TEST()
261 ix_t = ix_t.abs() % 1000; in TEST()
289 auto ix_t = ix.matrix<int64>(); in TEST() local
290 ix_t = ix_orig; in TEST()
304 ix_t = ix_orig; in TEST()
[all …]
Dgroup_iterator.cc24 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()
Dsparse_tensor.h276 auto ix_t = ix_.matrix<int64>(); in Reorder() local
286 FixedDimComparator<ORDER_SIZE> sorter(ix_t, order, shape()); \ in Reorder()
298 DimComparator sorter(ix_t, order, shape()); in Reorder()
321 std::swap_ranges(&(ix_t(n, 0)), &(ix_t(n + 1, 0)), &(ix_t(r, 0))); in Reorder()
365 auto ix_t = ix_.matrix<int64>(); in ToDense() local
366 const int64* const ix_ptr = ix_t.data(); in ToDense()
468 TTypes<int64>::Matrix ix_t = output_ix.matrix<int64>(); in Concat() local
481 auto* ix_out = &ix_t(offset, 0); in Concat()
Dgroup_iterator.h139 const auto& ix_t = iter_->ix_matrix_; in group_at() local
140 return ix_t(loc_, index); in group_at()
/external/tensorflow/tensorflow/core/util/
Dexample_proto_helper.cc159 auto ix_t = indices->matrix<int64>(); in CopyIntoSparseTensor() local
160 int64* ix_p = &ix_t(offset, 0); in CopyIntoSparseTensor()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbarrier_ops_test.py268 for ix_t, k_t, v_t in take_t:
269 index_fetches.append(ix_t)