Home
last modified time | relevance | path

Searched refs:indices_vec (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Ddynamic_stitch_op.cc202 auto indices_vec = indices_inputs[i].flat<int32>(); in Compute() local
204 for (int j = 0; j < indices_vec.size(); ++j) { in Compute()
208 indices_flat.Set(indices_vec(j), base_size + j); in Compute()
214 base_size += indices_vec.size(); in Compute()
257 auto indices_vec = indices.flat<int32>(); in Compute() local
260 data.shaped<T, 2>({indices_vec.dimension(0), slice_size}); in Compute()
265 for (int i = 0; i < indices_vec.size(); i++) { in Compute()
266 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
275 for (int i = 0; i < indices_vec.size(); i++) { in Compute()
278 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
Dgather_op_test.cc206 std::vector<Index> indices_vec; in Gather() local
207 indices_vec.reserve(kLookups); in Gather()
209 indices_vec.push_back(rnd.Uniform(kRows)); in Gather()
212 for (int i = 0; i < indices_vec.size(); i++) { in Gather()
213 indices.flat<Index>()(i) = indices_vec[i]; in Gather()
Dsegment_reduction_ops_impl.h478 const auto indices_vec = indices.vec<Index>();
558 const int bad_offset = Reduce<T, Index>(input_flat, indices_vec, start,
563 "] == ", indices_vec(start + bad_offset),
618 const typename TTypes<Tindex>::ConstVec& indices_vec, int64 start,
621 return ReduceImpl<Tin, Tindex, Tin>(input_flat, indices_vec, start, num,
628 const typename TTypes<Tindex>::ConstVec& indices_vec, int64 start,
632 ReduceImpl<Tin, Tindex, float>(input_flat, indices_vec, start, num,
641 const typename TTypes<Tindex>::ConstVec& indices_vec, int64 start,
645 const auto index##n = indices_vec(start + (i)); \
865 const auto indices_vec = indices.vec<Index>();
[all …]
Dtraining_ops.cc599 typename TTypes<Tindex>::ConstVec indices_vec, in operator ()()
601 const Tindex N = static_cast<Tindex>(indices_vec.dimension(0)); in operator ()()
616 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in operator ()()
650 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in operator ()()
1251 auto indices_vec = indices.vec<Tindex>(); in DoCompute() local
1253 const Tindex index = indices_vec(i); in DoCompute()
1269 const Tindex index = indices_vec(i); in DoCompute()
1443 auto indices_vec = indices.vec<Tindex>(); in Compute() local
1452 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
1476 auto indices_vec = indices.vec<Tindex>(); in Compute() local
[all …]
Dlist_kernels.h751 const auto indices_vec = indices.vec<int32>(); in Compute() local
755 : *std::max_element(indices_vec.data(), in Compute()
756 indices_vec.data() + indices.NumElements()); in Compute()
Dtraining_ops.h189 typename TTypes<Tindex>::ConstVec indices_vec,