HomeSort by relevance Sort by last modified time
    Searched refs:indices_vec (Results 1 - 5 of 5) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
dynamic_stitch_op.cc 202 auto indices_vec = indices_inputs[i].flat<int32>(); variable
204 for (int j = 0; j < indices_vec.size(); ++j) {
208 indices_flat.Set(indices_vec(j), base_size + j);
214 base_size += indices_vec.size();
256 auto indices_vec = indices.flat<int32>();
259 data.shaped<T, 2>({indices_vec.dimension(0), slice_size});
264 for (int i = 0; i < indices_vec.size(); i++) {
265 int32 index = internal::SubtleMustCopy(indices_vec(i));
274 for (int i = 0; i < indices_vec.size(); i++) {
277 int32 index = internal::SubtleMustCopy(indices_vec(i))
    [all...]
gather_op_test.cc 191 std::vector<Index> indices_vec; local
192 indices_vec.reserve(kLookups);
194 indices_vec.push_back(rnd.Uniform(kRows));
197 for (int i = 0; i < indices_vec.size(); i++) {
198 indices.flat<Index>()(i) = indices_vec[i];
segment_reduction_ops.cc 660 const auto indices_vec = indices.vec<Index>(); variable
735 Reduce(input_flat, indices_vec, start, end - start, out);
1034 const auto indices_vec = indices.vec<Index>(); variable
    [all...]
training_ops.cc 844 auto indices_vec = indices.vec<Tindex>(); local
1038 auto indices_vec = indices.vec<Tindex>(); variable
1071 auto indices_vec = indices.vec<Tindex>(); variable
1387 auto indices_vec = indices.vec<Tindex>(); variable
1410 auto indices_vec = indices.vec<Tindex>(); variable
1545 auto indices_vec = indices.vec<Tindex>(); variable
1580 auto indices_vec = indices.vec<Tindex>(); variable
1841 auto indices_vec = indices.vec<Tindex>(); variable
1877 auto indices_vec = indices.vec<Tindex>(); variable
2204 auto indices_vec = indices.vec<Tindex>(); variable
2273 auto indices_vec = indices.vec<Tindex>(); variable
2531 auto indices_vec = indices.vec<Tindex>(); variable
2750 auto indices_vec = indices.vec<Tindex>(); variable
3663 auto indices_vec = indices.vec<Tindex>(); variable
3801 auto indices_vec = indices.vec<Tindex>(); variable
    [all...]
list_kernels.h 774 const auto indices_vec = indices.vec<int32>(); variable
    [all...]

Completed in 3705 milliseconds