| /external/XNNPACK/test/ |
| D | constant-pad-operator-tester.h | 131 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestX8() local
246 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestX16() local
361 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestX32() local
|
| D | binary-elementwise-operator-tester.h | 234 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestQS8() local
416 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestQU8() local
591 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF16() local
812 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_strides; in TestF32() local
|
| /external/eigen/unsupported/test/ |
| D | cxx11_tensor_block_io.cpp | 40 const array<Index, NumDims>& output_strides) { in GetInputIndex()
86 auto output_strides = internal::strides<Layout>(dims); in test_block_io_copy_data_from_source_to_target() local
159 auto output_strides = internal::strides<Layout>(output_tensor_dims); in test_block_io_copy_using_reordered_dimensions() local
|
| D | cxx11_tensor_block_access.cpp | 156 const array<Index, NumDims>& output_strides) { in GetInputIndex()
199 const DSizes<StorageIndex, Dim> &input_dims, &input_strides, &output_dims, &output_strides; member in EqualityChecker
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | reshape_util.cc | 59 gtl::InlinedVector<int64_t, 8> output_strides(output_rank); in operator ()() local
|
| D | edit_distance_op.cc | 179 std::vector<int64_t> output_strides(output_shape.dims()); in Compute() local
|
| D | sparse_reduce_op.cc | 200 gtl::InlinedVector<int64_t, 8> output_strides(reduction.group_by_dims.size()); in Compute() local
|
| D | set_kernels.cc | 271 const auto output_strides = Strides(output_shape); in Compute() local
|
| /external/libaom/av1/encoder/ |
| D | cnn.h | 162 const int *output_strides; member
|
| D | cnn.c | 1207 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img() local
1225 const int output_strides[1] = { out_stride }; in av1_cnn_predict_img_highbd() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_slice_op.cc | 446 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_evaluator.cc | 1644 const auto output_strides = in ComputeFft() local
2041 const absl::Span<const int64_t> output_strides, in CopyDataToOutput()
2098 const absl::Span<const int64_t> output_strides, in CopyDataToOutput()
|
| /external/libaom/test/ |
| D | cnn_test.cc | 63 const int output_strides[1] = { out_stride }; in RunCNNTest() local
|
| /external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3538 std::vector<int64_t> output_strides = output_descriptor.vectorized_strides( in GetCudnnOperationGraph() local
3672 std::vector<int64_t> output_strides = output_descriptor.vectorized_strides( in GetCudnnFusedOperationGraph() local
|