| /aosp_15_r20/external/XNNPACK/test/ |
| H A D | constant-pad-operator-tester.h | 130 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX8() local
245 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX16() local
360 std::array<size_t, XNN_MAX_TENSOR_DIMS> input_strides; in TestX32() local
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_block_io.cpp | 39 const array<Index, NumDims>& input_strides, in GetInputIndex()
85 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_data_from_source_to_target() local
158 auto input_strides = internal::strides<Layout>(dims); in test_block_io_copy_using_reordered_dimensions() local
|
| H A D | cxx11_tensor_block_access.cpp | 155 const array<Index, NumDims>& input_strides, in GetInputIndex()
199 const DSizes<StorageIndex, Dim> &input_dims, &input_strides, &output_dims, &output_strides; member in EqualityChecker
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | LinearAlgebraUtils.h | 582 IntArrayRef input_strides = input.strides(); in is_blas_compatible_column_major_order() local
603 IntArrayRef input_strides = input.strides(); in is_blas_compatible_row_major_order() local
|
| H A D | SpectralOps.cpp | 1200 const auto input_strides = input.strides(); in _fft_fill_with_conjugate_symmetry_() local
|
| /aosp_15_r20/external/executorch/kernels/aten/cpu/util/ |
| H A D | copy_ops_util.cpp | 58 auto input_strides = input.strides(); in check__to_dim_order_copy_args() local
|
| /aosp_15_r20/external/executorch/kernels/aten/cpu/ |
| H A D | op__to_dim_order_copy.cpp | 82 auto input_strides = input.strides(); in check__to_dim_order_copy_args() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | reshape_util.cc | 51 gtl::InlinedVector<int64_t, 8> input_strides(input_rank); in operator ()() local
|
| H A D | set_kernels.cc | 216 const VarDimArray& input_strides, in PopulateFromDenseGroup()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| H A D | resolve_constant_transpose.cc | 59 int input_strides[4]; in Transpose() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/cuda/ |
| H A D | CUDASparseDescriptors.cpp | 61 IntArrayRef input_strides = input.strides(); in createRawDnMatDescriptor() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | IndexKernel.cpp | 662 auto input_strides = iter.strides(1); in cpu_hflip_channels_last_vec() local
719 auto input_strides = iter.strides(1); in flip_kernel() local
|
| /aosp_15_r20/external/executorch/kernels/quantized/cpu/ |
| H A D | op_dequantize.cpp | 423 std::array<exec_aten::StridesType, 2> input_strides; in dequantize_per_token_out() local
|
| H A D | op_quantize.cpp | 406 std::array<exec_aten::StridesType, 2> input_strides; in quantize_per_token_out() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_slice_op.cc | 407 memory::dims input_dims, input_strides; in ComputeMklSlice() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | hlo_evaluator.cc | 1643 const auto input_strides = ComputeStrides(input_lengths, input_literal); in ComputeFft() local
1988 const absl::Span<const int64_t> input_strides, in CopyDataFromInput()
2078 const absl::Span<const int64_t> input_strides, in CopyDataFromInput()
|
| /aosp_15_r20/external/executorch/extension/pybindings/ |
| H A D | pybindings.cpp | 682 input_strides; in run_method() local
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_ops.py | 2037 def check_inplace_view(func, input, rs, input_size, input_strides): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | mkl_util.h | 1244 dnnl_dim_t* input_strides = new dnnl_dim_t[kNumDims]; in CreateBlockedMemDescHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/toco/ |
| H A D | tooling_util.cc | 2153 int input_strides[4]; in ShuffleArrayTemplate() local
|
| /aosp_15_r20/external/pytorch/torch/_dynamo/ |
| H A D | symbolic_convert.py | 246 input_strides: Dict[str, List[int]] = dataclasses.field(default_factory=dict) variable in LocalState
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/ |
| H A D | cuda_dnn.cc | 3521 std::vector<int64_t> input_strides = input_descriptor.vectorized_strides( in GetCudnnOperationGraph() local
3656 std::vector<int64_t> input_strides = input_descriptor.vectorized_strides( in GetCudnnFusedOperationGraph() local
|