Searched refs:indices_vec (Results 1 – 6 of 6) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | dynamic_stitch_op.cc | 202 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()
|
D | gather_op_test.cc | 206 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()
|
D | segment_reduction_ops_impl.h | 478 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 …]
|
D | training_ops.cc | 599 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 …]
|
D | list_kernels.h | 751 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()
|
D | training_ops.h | 189 typename TTypes<Tindex>::ConstVec indices_vec,
|