OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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