| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | maxpooling_op_gpu.cu.cc | 71 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in MaxPoolForwardNCHW()
115 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNoMaskKernel_NCHW_VECT_C()
148 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in MaxPoolForwardNHWC()
234 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNCHW()
275 const int width, const int kernel_h, const int kernel_w, const int stride_h, in MaxPoolGradBackwardNoMaskNHWC()
358 const int stride_h, const int stride_w, const int pad_t, const int pad_l, in operator ()()
377 const int kernel_h, const int kernel_w, const int stride_h, in operator ()()
428 const int kernel_w, const int stride_h, const int stride_w, const int pad_t, in operator ()()
|
| H A D | pooling_ops_3d_gpu.cu.cc | 36 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNCDHW()
87 const int stride_h, const int stride_w, const int pad_p, const int pad_t, in MaxPoolGradBackwardNoMaskNDHWC()
142 const int kernel_w, const int stride_p, const int stride_h, in operator ()()
|
| H A D | xsmm_conv2d_test.cc | 47 int stride_h; member
131 int stride_h = param->stride_h; in naive_conv_fp() local
201 int stride_h = stride; in TEST_F() local
|
| H A D | conv_grad_filter_ops.cc | 77 const int stride_h, const int stride_w, T* col_data) { in Im2col()
276 int stride_h = GetTensorDim(strides_, data_format_, 'H'); in Conv2DBackpropFilterOp() local
|
| H A D | avgpooling_op_gpu.cu.cc | 86 const int kernel_w, const int stride_h, in RunAvePoolBackwardNHWC()
|
| H A D | conv_grad_input_ops.h | 81 const int stride_h, const int stride_w, T* __restrict im_data) { in Col2im()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/ |
| H A D | Unfold3d.cpp | 192 int64_t stride_h, in Unfold3dZeroPaddingCopyKernelImpl()
236 int64_t stride_h, in Unfold3dCopyKernelImpl()
315 int64_t stride_h, in Unfold3dZeroPaddingAccKernelImpl()
368 int64_t stride_h, in Unfold3dAccKernelImpl()
449 int64_t stride_h, in Unfold3dCopyCPU()
497 int64_t stride_h, in Unfold3dAccCPU()
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/ |
| H A D | vol2col.cuh | 31 const int stride_h, in vol2col_kernel()
92 const int stride_h, in vol2col()
143 const unsigned stride_h, in vol2im_kernel()
216 const int64_t stride_h, in col2vol()
|
| H A D | AveragePool2d.cu | 37 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_out_cuda_frame()
86 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in avg_pool2d_out_cuda_frame_nhwc()
134 const int kernel_h, const int kernel_w, const int stride_h, in avg_pool2d_backward_out_cuda_frame()
191 const int kernel_h, const int kernel_w, const int stride_h, in avg_pool2d_backward_out_cuda_frame_nhwc()
|
| H A D | DilatedMaxPool2d.cu | 48 const int kernel_h, const int kernel_w, const int stride_h, in max_pool_forward_nchw()
181 const int stride_h, const int stride_w, const int pad_h, const int pad_w, in max_pool_backward_nchw()
|
| /aosp_15_r20/external/pytorch/test/ |
| H A D | test_xnnpack_integration.py | 109 stride_h, argument
178 stride_h, argument
332 stride_h, argument
447 stride_h, argument
601 stride_h, argument
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/ |
| H A D | PixelShuffleKernel.cpp | 37 int64_t stride_h = width; in cpu_pixel_shuffle() local
133 int64_t stride_h = S * width * S; in cpu_pixel_unshuffle() local
174 int64_t stride_h = S * width * S * sub_channels; in cpu_pixel_unshuffle_channels_last() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfr/examples/mnist/ |
| H A D | ops_defs.py | 53 def _composite_conv_add_relu(input_, filter_, bias, stride_w, stride_h, argument
172 def _composite_max_pool(input_, stride_w, stride_h, filter_width, filter_height, argument
|
| /aosp_15_r20/external/pytorch/test/quantization/core/ |
| H A D | test_quantized_op.py | 5154 stride_h, argument
5235 stride_h, argument
5442 stride_h, argument
5525 stride_h, argument
5782 stride_h, argument
5915 stride_h, argument
6407 stride_h, argument
6491 stride_h, argument
6560 self, inputs, stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, o_pad, argument
|
| H A D | test_quantized_functional.py | 141 groups, kernel_h, kernel_w, stride_h, stride_w, pad_h, pad_w, dilation, argument
198 stride_d, stride_h, stride_w, pad_d, pad_h, pad_w, dilation, X_scale, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/quantization/tensorflow/utils/ |
| H A D | tf_to_xla_attribute_utils.cc | 94 const int stride_h = strides[1].cast<IntegerAttr>().getInt(); in PadForDynamicShapedInputSamePadding() local
170 const int stride_h = strides[1].cast<IntegerAttr>().getInt(); in CalculatePaddingAndPadIfNeeded() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | transpose_conv_test.cc | 62 Padding padding, int stride_w, int stride_h, in BaseTransposeConvOpModel()
588 int stride_w, int stride_h, TestType test_type, in BaseTransposeConvBiasOpModel()
|
| /aosp_15_r20/external/pytorch/test/quantization/eager/ |
| H A D | test_quantize_eager_qat.py | 768 stride_h, argument
920 stride_h, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| H A D | legalize_tf.cc | 521 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
591 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
878 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
971 int64_t stride_h = tmpAttr[1].dyn_cast<IntegerAttr>().getInt(); in matchAndRewrite() local
|
| H A D | legalize_tfl.cc | 892 int64_t stride_h = tfl_avgpool_op.stride_h(); in matchAndRewrite() local
969 int64_t stride_h = tfl_maxpool_op.stride_h(); in matchAndRewrite() local
1032 int64_t stride_h = tfl_conv2d_op.stride_h(); in matchAndRewrite() local
1124 int64_t stride_h = tfl_conv_op.stride_h(); in matchAndRewrite() local
1264 int64_t stride_h = tfl_conv2d_op.stride_h(); in matchAndRewrite() local
|
| /aosp_15_r20/external/executorch/kernels/portable/cpu/ |
| H A D | op_pixel_unshuffle.cpp | 39 const auto stride_h = width; in pixel_unshuffle_impl() local
|
| H A D | op_pixel_shuffle.cpp | 36 const auto stride_h = width; in pixel_shuffle_impl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/ |
| H A D | optional_input.json | 72 "stride_h": 1 number
|
| H A D | import_json.json | 77 "stride_h": 1 number
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/ |
| H A D | XnnpackUtils.h | 47 uint32_t stride_h, in xnnp_create_convolution2d_nhwc()
|