| /aosp_15_r20/external/XNNPACK/test/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A 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
|
| H A 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
|
| /aosp_15_r20/external/pytorch/torch/csrc/profiler/standalone/ |
| H A D | execution_trace_observer.cpp | 238 const std::string& output_strides = "[]", in writeJsonNode()
711 std::vector<std::string> output_strides; in onFunctionExit() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | reshape_util.cc | 59 gtl::InlinedVector<int64_t, 8> output_strides(output_rank); in operator ()() local
|
| H A D | edit_distance_op.cc | 179 std::vector<int64_t> output_strides(output_shape.dims()); in Compute() local
|
| H A D | sparse_reduce_op.cc | 200 gtl::InlinedVector<int64_t, 8> output_strides(reduction.group_by_dims.size()); in Compute() local
|
| H A D | set_kernels.cc | 271 const auto output_strides = Strides(output_shape); in Compute() local
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | cnn.h | 162 const int *output_strides; member
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_slice_op.cc | 446 auto output_strides = CalculateTFStrides(size_dims); in ComputeMklSlice() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | symbolic_shape_runtime_fusion.cpp | 345 std::vector<StrideInput>& output_strides) { in insertDynamicShapesGuard()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A 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()
|
| /aosp_15_r20/external/libaom/test/ |
| H A D | cnn_test.cc | 63 const int output_strides[1] = { out_stride }; in RunCNNTest() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | IndexKernel.cpp | 718 auto output_strides = iter.strides(0); in flip_kernel() local
|
| /aosp_15_r20/external/pytorch/test/cpp/tensorexpr/ |
| H A D | test_external_calls.cpp | 942 at::Device device) { in TEST()
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A 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
|