Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/cpu/kernels/
DCpuWinogradConv2dKernel.cpp51 const size_t input_batch_stride = src_strides[batch_idx] / element_size_in_bytes; in run_op() local
/external/XNNPACK/src/xnnpack/
Dcompute.h463 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 …]
/external/XNNPACK/src/operators/
Dconvolution-nchw.c581 …const size_t input_batch_stride = (input_height * input_width * convolution_op->input_pixel_stride… in setup_convolution2d_nchw() local
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
Ddepthwise_conv_hybrid.h142 const int input_batch_stride = input_height_stride * input_shape.Dims(1); in DepthwiseConvHybridGeneral() local