input_batch_stride (definition) in projects: aosp_15_r20

Searched defs:input_batch_stride (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/XNNPACK/src/xnnpack/
H A D	compute.h	`463 size_t input_batch_stride; member 493 size_t input_batch_stride; member 527 size_t input_batch_stride; member 552 size_t input_batch_stride; member 602 size_t input_batch_stride; member 636 size_t input_batch_stride; member 673 size_t input_batch_stride; member 712 size_t input_batch_stride; member 743 size_t input_batch_stride; member 769 size_t input_batch_stride; member [all …]`
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A D	CpuWinogradConv2dKernel.cpp	`51 const size_t input_batch_stride = src_strides[batch_idx] / element_size_in_bytes; in run_op() local`
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A D	Col2Im.cu	`100 int64_t input_batch_stride = input.stride(0); in col2im_out_cuda_template() local`
/aosp_15_r20/external/XNNPACK/src/operators/
H A D	convolution-nchw.c	`581 …const size_t input_batch_stride = (input_height * input_width * convolution_op->input_pixel_stride… in setup_convolution2d_nchw() local`
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A D	depthwise_conv_hybrid.h	`142 const int input_batch_stride = input_height_stride * input_shape.Dims(1); in DepthwiseConvHybridGeneral() local`
/aosp_15_r20/external/pytorch/torch/sparse/
H A D	_triton_ops.py	`2267 input_batch_stride, argument`
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/qnnpack/src/
H A D	operator-run.c	`645 size_t input_batch_stride; member`