Home
last modified time | relevance | path

Searched refs:indices_vec (Results 1 – 5 of 5) 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()
256 auto indices_vec = indices.flat<int32>(); in Compute() local
259 data.shaped<T, 2>({indices_vec.dimension(0), slice_size}); in Compute()
264 for (int i = 0; i < indices_vec.size(); i++) { in Compute()
265 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
274 for (int i = 0; i < indices_vec.size(); i++) { in Compute()
277 int32 index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
Dgather_op_test.cc191 std::vector<Index> indices_vec; in Gather() local
192 indices_vec.reserve(kLookups); in Gather()
194 indices_vec.push_back(rnd.Uniform(kRows)); in Gather()
197 for (int i = 0; i < indices_vec.size(); i++) { in Gather()
198 indices.flat<Index>()(i) = indices_vec[i]; in Gather()
Dtraining_ops.cc844 auto indices_vec = indices.vec<Tindex>(); in DoCompute() local
846 const Tindex index = indices_vec(i); in DoCompute()
862 const Tindex index = indices_vec(i); in DoCompute()
1038 auto indices_vec = indices.vec<Tindex>(); in Compute() local
1047 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
1071 auto indices_vec = indices.vec<Tindex>(); in Compute() local
1080 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
1387 auto indices_vec = indices.vec<Tindex>(); in Compute() local
1396 const Tindex index = internal::SubtleMustCopy(indices_vec(i)); in Compute()
1410 auto indices_vec = indices.vec<Tindex>(); in Compute() local
[all …]
Dsegment_reduction_ops.cc660 const auto indices_vec = indices.vec<Index>(); in Compute() local
735 Reduce(input_flat, indices_vec, start, end - start, out); in Compute()
739 "] == ", indices_vec(start + bad_offset), in Compute()
763 const typename TTypes<Index>::ConstVec& indices_vec, int64 start, in Reduce() argument
767 const auto index##n = indices_vec(start + (i)); \ in Reduce()
1034 const auto indices_vec = indices.vec<Index>(); in Compute() local
1074 const Index output_idx = internal::SubtleMustCopy(indices_vec(i)); in Compute()
Dlist_kernels.h774 const auto indices_vec = indices.vec<int32>(); in Compute() local
778 : *std::max_element(indices_vec.data(), in Compute()
779 indices_vec.data() + indices.NumElements()); in Compute()