Home
last modified time | relevance | path

Searched refs:src_strides (Results 1 – 13 of 13) sorted by relevance

/external/ComputeLibrary/src/cpu/kernels/
DCpuWinogradConv2dKernel.cpp47 const auto src_strides = input_nhwc->info()->strides_in_bytes(); in run_op() local
49 const size_t input_row_stride = src_strides[height_idx] / element_size_in_bytes; in run_op()
50 const size_t input_col_stride = src_strides[width_idx] / element_size_in_bytes; in run_op()
51 const size_t input_batch_stride = src_strides[batch_idx] / element_size_in_bytes; in run_op()
/external/executorch/backends/apple/coreml/runtime/test/
DMultiArrayTests.mm78 const std::vector<ssize_t>& src_strides,
81 auto src_multiarray = make_multi_array_and_fill<T1>(shape, src_strides, src_storage);
96 const std::vector<ssize_t>& src_strides,
98 verify_copy_<T1, T2>(shape, src_strides, dst_strides);
99 verify_copy_<T2, T1>(shape, src_strides, dst_strides);
/external/libvpx/vp9/encoder/
Dvp9_frame_scale.c28 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; in vp9_scale_and_extend_frame_c() local
79 const int src_stride = src_strides[i]; in vp9_scale_and_extend_frame_c()
116 const int src_stride = src_strides[i]; in vp9_scale_and_extend_frame_c()
Dvp9_encoder.c3094 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; local
3110 src_strides[i], dsts[i], dst_heights[i],
3113 vp9_resize_plane(srcs[i], src_heights[i], src_widths[i], src_strides[i],
3117 vp9_resize_plane(srcs[i], src_heights[i], src_widths[i], src_strides[i],
3152 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; local
3160 const int src_stride = src_strides[i];
/external/pytorch/aten/src/ATen/native/cuda/
DScatterGatherKernel.cu171 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local
183 src.as_strided(index_sizes, src_strides) in operator ()()
229 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local
241 src.as_strided(index_sizes, src_strides) in operator ()()
288 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local
300 src.as_strided(index_sizes, src_strides) in operator ()()
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_eltwise_activation_base_op.h272 auto src_strides = CalculateTFStrides(src_dims); in Compute() local
275 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
Dmkl_relu_op.cc472 auto src_strides = CalculateTFStrides(src_dims); in Compute() local
474 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
629 auto src_strides = CalculateTFStrides(src_dims); in Compute() local
630 src_md = MklDnnData<T>::CreateBlockedMemDesc(src_dims, src_strides); in Compute()
/external/cronet/tot/third_party/libc++/src/test/std/containers/views/mdspan/layout_stride/
Dcomparison.pass.cpp73 std::array<int, From::rank()> src_strides) { in test_comparison() argument
75 std::layout_stride::mapping<From> src(src_exts, src_strides); in test_comparison()
/external/cronet/stable/third_party/libc++/src/test/std/containers/views/mdspan/layout_stride/
Dcomparison.pass.cpp73 std::array<int, From::rank()> src_strides) { in test_comparison() argument
75 std::layout_stride::mapping<From> src(src_exts, src_strides); in test_comparison()
/external/libvpx/vp9/common/
Dvp9_postproc.c239 const int src_strides[3] = { src->y_stride, src->uv_stride, in vp9_deblock() local
251 src_strides[i], dst_strides[i], src_heights[i], src_widths[i], ppl); in vp9_deblock()
/external/pytorch/aten/src/ATen/native/mps/operations/
DView.mm818 std::vector<uint32_t> src_strides(kernel_size == 0 ? 1 : kernel_size);
821 src_sizes[0] = src_strides[0] = 1;
825 src_strides[i] = (uint32_t)(src.strides()[i]);
833 mtl_setBytes(computeEncoder, src_strides, 3);
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorBlock.h1164 Src(const Dimensions& src_strides, const Scalar* src,
1166 : strides(src_strides), data(src), offset(src_offset) {}
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_evaluator.cc1941 const absl::Span<const int64_t> src_strides, in GenerateIndices() argument
1946 CHECK_EQ(src_lengths.size() + 1, src_strides.size()); in GenerateIndices()
1960 src_index += src_strides[axis]; in GenerateIndices()