HomeSort by relevance Sort by last modified time
    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