Searched defs:src_strides (Results 1 – 9 of 9) sorted by relevance
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
H A D | ScatterGatherKernel.cu | 171 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local 229 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local 288 auto src_strides = ensure_nonempty_vec(src.strides().vec()); in operator ()() local
|
/aosp_15_r20/external/cronet/third_party/libc++/src/test/std/containers/views/mdspan/layout_stride/ |
H A D | comparison.pass.cpp | 72 std::array<int, From::rank()> src_strides) { in test_comparison()
|
/aosp_15_r20/external/libvpx/vp9/encoder/ |
H A D | vp9_frame_scale.c | 28 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; in vp9_scale_and_extend_frame_c() local
|
H A D | vp9_encoder.c | 3094 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; local 3152 const int src_strides[3] = { src->y_stride, src->uv_stride, src->uv_stride }; local
|
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/ |
H A D | CpuWinogradConv2dKernel.cpp | 47 const auto src_strides = input_nhwc->info()->strides_in_bytes(); in run_op() local
|
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_eltwise_activation_base_op.h | 272 auto src_strides = CalculateTFStrides(src_dims); in Compute() local
|
H A D | mkl_relu_op.cc | 472 auto src_strides = CalculateTFStrides(src_dims); in Compute() local 629 auto src_strides = CalculateTFStrides(src_dims); in Compute() local
|
/aosp_15_r20/external/libvpx/vp9/common/ |
H A D | vp9_postproc.c | 239 const int src_strides[3] = { src->y_stride, src->uv_stride, in vp9_deblock() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | hlo_evaluator.cc | 1941 const absl::Span<const int64_t> src_strides, in GenerateIndices()
|