OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
Searched
refs:output_strides
(Results
1 - 5
of
5
) sorted by null
/external/tensorflow/tensorflow/core/kernels/
reshape_util.cc
115
gtl::InlinedVector<int64, 8>
output_strides
(output_rank);
116
output_strides
[output_rank - 1] = 1;
118
output_strides
[d] =
output_strides
[d + 1] * output_shape.dim_size(d + 1);
134
output_ind(i, j) = id /
output_strides
[j];
135
id %=
output_strides
[j];
edit_distance_op.cc
157
std::vector<int64>
output_strides
(output_shape.dims());
158
output_strides
[output_shape.dims() - 1] = 1;
160
output_strides
[d] =
output_strides
[d + 1] * output_shape.dim_size(d + 1);
182
output_strides
.begin(), int64{0});
191
output_strides
.begin(), int64{0});
199
output_strides
.begin(), int64{0});
209
output_strides
.begin(), int64{0});
221
output_strides
.begin(), int64{0});
sparse_reduce_op.cc
193
gtl::InlinedVector<int64, 8>
output_strides
(reduction.group_by_dims.size());
194
if (!
output_strides
.empty()) { // Do this iff we don't reduce all.
195
output_strides
.back() = 1;
196
for (int d =
output_strides
.size() - 2; d >= 0; --d) {
197
output_strides
[d] =
198
output_strides
[d + 1] * shape_vec(reduction.group_by_dims[d + 1]);
220
const int64 idx = CoordinatesToFlatIndex(g.group(),
output_strides
);
mkl_reshape_op.cc
278
auto
output_strides
= CalculateTFStrides(output_dims);
variable
280
output_dims,
output_strides
);
set_kernels.cc
263
const auto
output_strides
= Strides(output_shape);
local
282
group_key.begin(), group_key.end(),
output_strides
.begin(), 0LL);
Completed in 93 milliseconds