Searched refs:output_strides (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | edit_distance_op.cc | 162 std::vector<int64> output_strides(output_shape.dims()); in Compute() local
163 output_strides[output_shape.dims() - 1] = 1; in Compute()
165 output_strides[d] = output_strides[d + 1] * output_shape.dim_size(d + 1); in Compute()
187 output_strides.begin(), int64{0}); in Compute()
196 output_strides.begin(), int64{0}); in Compute()
204 output_strides.begin(), int64{0}); in Compute()
214 output_strides.begin(), int64{0}); in Compute()
226 output_strides.begin(), int64{0}); in Compute()
|
| D | reshape_util.cc | 117 gtl::InlinedVector<int64, 8> output_strides(output_rank); in Reshape() local
119 output_strides[output_rank - 1] = 1; in Reshape()
121 output_strides[d] = output_strides[d + 1] * output_shape.dim_size(d + 1); in Reshape()
138 output_ind(i, j) = id / output_strides[j]; in Reshape()
139 id %= output_strides[j]; in Reshape()
|
| D | sparse_reduce_op.cc | 195 gtl::InlinedVector<int64, 8> output_strides(reduction.group_by_dims.size()); in Compute() local
196 if (!output_strides.empty()) { // Do this iff we don't reduce all. in Compute()
197 output_strides.back() = 1; in Compute()
198 for (int d = output_strides.size() - 2; d >= 0; --d) { in Compute()
199 output_strides[d] = in Compute()
200 output_strides[d + 1] * shape_vec(reduction.group_by_dims[d + 1]); in Compute()
222 const int64 idx = CoordinatesToFlatIndex(g.group(), output_strides); in Compute()
|
| D | mkl_reshape_op.cc | 179 auto output_strides = CalculateTFStrides(output_dims); in Compute() local
181 output_dims, output_strides); in Compute()
|
| D | mkl_slice_op.cc | 415 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice() local
417 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
|
| D | set_kernels.cc | 260 const auto output_strides = Strides(output_shape); in Compute() local
279 group_key.begin(), group_key.end(), output_strides.begin(), 0LL); in Compute()
|