| /external/XNNPACK/test/ |
| D | binary-elementwise-operator-tester.h | 235 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestQS8() local
240 output_strides[i - 1] = output_stride; in TestQS8()
262 …_ref[i * output_strides[0] + j * output_strides[1] + k * output_strides[2] + l * output_strides[3]… in TestQS8()
329 …i * output_strides[0] + j * output_strides[1] + k * output_strides[2] + l * output_strides[3] + m … in TestQS8()
373 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF16() local
378 output_strides[i - 1] = output_stride; in TestF16()
400 …_ref[i * output_strides[0] + j * output_strides[1] + k * output_strides[2] + l * output_strides[3]… in TestF16()
484 …i * output_strides[0] + j * output_strides[1] + k * output_strides[2] + l * output_strides[3] + m … in TestF16()
524 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF32() local
529 output_strides[i - 1] = output_stride; in TestF32()
[all …]
|
| D | constant-pad-operator-tester.h | 131 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestX32() local
135 output_strides[i - 1] = output_stride; in TestX32()
157 (i + input_pre_paddings[0]) * output_strides[0] + in TestX32()
158 (j + input_pre_paddings[1]) * output_strides[1] + in TestX32()
159 (k + input_pre_paddings[2]) * output_strides[2] + in TestX32()
160 (l + input_pre_paddings[3]) * output_strides[3] + in TestX32()
161 (m + input_pre_paddings[4]) * output_strides[4] + in TestX32()
162 (n + input_pre_paddings[5]) * output_strides[5]; in TestX32()
205 i * output_strides[0] + j * output_strides[1] + k * output_strides[2] + in TestX32()
206 l * output_strides[3] + m * output_strides[4] + n * output_strides[5]; in TestX32()
|
| /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 | 55 gtl::InlinedVector<int64, 8> output_strides(output_rank); in operator ()() local
57 output_strides[output_rank - 1] = 1; in operator ()()
59 output_strides[d] = in operator ()()
60 output_strides[d + 1] * output_shape.dim_size(d + 1); in operator ()()
70 output_indices(i, j) = id / output_strides[j]; in operator ()()
71 id %= output_strides[j]; in operator ()()
|
| 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 | 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()
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | ceva_tflm_lib.h | 103 const int* output_strides, const int* output_extents);
108 const int* strides2, const int* output_strides);
118 const int* strides1, const int* strides2, const int* output_strides,
122 const int* strides1, const int* strides2, const int* output_strides,
198 const int* strides1, const int* strides2, const int* output_strides,
202 const int* strides1, const int* strides2, const int* output_strides,
215 const int* strides2, const int* output_strides, const int* output_extents);
220 const int* strides1, const int* strides2, const int* output_strides);
|
| /external/libaom/libaom/av1/encoder/ |
| D | cnn.h | 161 const int *output_strides; member
|
| D | cnn.c | 854 const int *out_stride = output_struct->output_strides; in av1_cnn_predict_c()
1120 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img() local
1122 .output_strides = output_strides, in av1_cnn_predict_img()
1138 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img_highbd() local
1140 .output_strides = output_strides, in av1_cnn_predict_img_highbd()
|
| D | partition_strategy.c | 99 .output_strides = output_dims, in av1_intra_mode_cnn_partition()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1188 const absl::Span<const int64> output_strides, in CopyDataToOutput() argument
1209 output_data[dst_index + i * output_strides[axis]] = value; in CopyDataToOutput()
1215 GenerateIndices(output_lengths, output_strides, fft_lengths, fft_strides, in CopyDataToOutput()
1245 const absl::Span<const int64> output_strides, in CopyDataToOutput() argument
1250 fft_strides, output_lengths, output_strides, in CopyDataToOutput()
1255 output_strides, output_literal); in CopyDataToOutput()
1413 const auto output_strides = ComputeStrides(output_lengths, output_literal); in HandleFft() local
1435 output_strides, &output_literal); in HandleFft()
1440 GenerateIndices(output_lengths, output_strides, input_lengths, in HandleFft()
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 439 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice() local
441 MklDnnData<T>::CreateBlockedMemDesc(size_dims, output_strides); in ComputeMklSlice()
|
| /external/libaom/libaom/test/ |
| D | cnn_test.cc | 54 const int output_strides[1] = { out_stride }; in RunCNNTest() local
55 CNN_MULTI_OUT output_struct = { num_outputs, output_chs, output_strides, in RunCNNTest()
|