| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | eigen_backward_spatial_convolutions_test.cc | 33 const int input_cols = 4; in TEST() local
82 const int input_cols = 4; in TEST() local
134 const int input_cols = 9; in TEST() local
182 const int input_cols = 9; in TEST() local
234 const int input_cols = 9; in TEST() local
289 const int input_cols = 9; in TEST() local
342 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid()
401 int input_cols = 5; in TEST() local
418 const int input_cols, const int output_depth) { in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
478 int input_cols = 5; in TEST() local
[all …]
|
| H A D | eigen_pooling_test.cc | 30 const int input_cols = 5; in TEST() local
78 const int input_cols = 5; in TEST() local
128 const int input_cols = 5; in TEST() local
188 const int input_cols = 5; in TEST() local
248 const int input_cols = 5; in TEST() local
310 const int input_cols = 5; in TEST() local
372 const int input_cols = 5; in TEST() local
450 const int input_cols = 5; in TEST() local
527 const int input_cols = 5; in TEST() local
575 const int input_cols = 5; in TEST() local
[all …]
|
| H A D | eigen_backward_cuboid_convolutions_test.cc | 34 const int input_cols = 4; in TEST() local
97 const int input_cols = 4; in TEST() local
160 const int input_cols = 4; in TEST() local
231 const int input_cols = 4; in TEST() local
303 const int input_cols = 4; in TEST() local
370 const int input_cols = 4; in TEST() local
436 const int input_cols = 4; in TEST() local
502 const int input_cols = 4; in TEST() local
569 const int input_cols = 9; in TEST() local
636 const int input_cols = 9; in TEST() local
[all …]
|
| H A D | dilation_ops_gpu.cu.cc | 40 const T* __restrict__ filter_ptr, int batch, int input_rows, int input_cols, in DilationKernel()
80 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropInputKernel()
130 int batch, int input_rows, int input_cols, int depth, int filter_rows, in DilationBackpropFilterKernel()
187 const int input_cols = input.dimension(2); in operator ()() local
218 const int input_cols = input.dimension(2); in operator ()() local
260 const int input_cols = input.dimension(2); in operator ()() local
|
| H A D | eigen_spatial_convolutions_test.cc | 32 const int input_cols = 5; in TEST() local
77 const int input_cols = 5; in TEST() local
196 const int input_cols = 5; in TEST() local
248 const int input_cols = 5; in TEST() local
306 const int input_cols = 5; in TEST() local
362 const int input_cols = 5; in TEST() local
410 const int input_cols = 3; in TEST() local
458 const int input_cols = 5; in TEST() local
509 const int input_cols = 7; in TEST() local
560 const int input_cols = 7; in TEST() local
[all …]
|
| H A D | dilation_ops.cc | 77 const int input_cols = input.dim_size(2); in ParseSizes() local
172 const int input_cols = input.dimension(2); in operator ()() local
287 const int input_cols = input.dimension(2); in operator ()() local
415 const int input_cols = input.dimension(2); in operator ()() local
|
| H A D | conv_ops.cc | 368 int input_cols, int in_depth, int filter_rows, in Run()
384 int input_cols, int in_depth, int filter_rows, in Run()
425 int input_cols, int in_depth, int filter_rows, in Run()
439 int input_cols, int in_depth, int filter_rows, in Run()
597 const int input_cols = static_cast<int>(input_cols_raw); in ComputeConv2DDimension() local
|
| H A D | eigen_benchmark.h | 88 Eigen::Index input_cols = input_dims[2]; in SpatialConvolutionBackwardInput() local
221 Eigen::Index input_cols = input_dims[2]; in CuboidConvolutionBackwardInput() local
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/ |
| H A D | input_transform.hpp | 145 TransformBase(const std::string &name, unsigned int input_rows, unsigned int input_cols) in TransformBase()
212 …TransformDirect(const std::string &name, unsigned int input_rows, unsigned int input_cols, Kernel … in TransformDirect()
283 …TransformIndirect(const std::string &name, unsigned int input_rows, unsigned int input_cols, Kerne… in TransformIndirect()
362 …TransformUnpadded(const std::string &name, unsigned int input_rows, unsigned int input_cols, Kerne… in TransformUnpadded()
|
| /aosp_15_r20/external/eigen/unsupported/test/ |
| H A D | cxx11_tensor_image_patch.cpp | 187 int input_cols = 3; in test_patch_padding_valid() local
263 int input_cols = 5; in test_patch_padding_valid_same_value() local
336 int input_cols = 2; in test_patch_padding_same() local
413 int input_cols = 1; in test_patch_padding_same_negative_padding_clip_to_zero() local
|
| H A D | cxx11_tensor_image_patch_sycl.cpp | 262 IndexType input_cols = 3; in test_patch_padding_valid_sycl() local
368 IndexType input_cols = 5; in test_patch_padding_valid_same_value_sycl() local
465 IndexType input_cols = 2; in test_patch_padding_same_sycl() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| H A D | runtime_conv2d_acl.cc | 31 int64_t input_batch, int64_t input_rows, int64_t input_cols, in ACLDepthwiseConvImpl()
129 int64_t input_cols, int64_t input_channels, in ACLGemmConvImpl()
235 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_ACLConv2DF32()
|
| H A D | runtime_single_threaded_conv2d.cc | 25 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
44 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
|
| H A D | runtime_conv2d.cc | 27 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenConv2DF32()
50 int64_t input_cols, int64_t input_channels, int64_t kernel_rows, in __xla_cpu_runtime_EigenConv2DF16()
|
| H A D | runtime_conv2d_mkl.cc | 54 int64_t input_cols, int64_t input_channels, in MKLConvImpl()
156 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_MKLConv2DF32()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_conv_ops.h | 141 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
164 int input_cols = static_cast<int>(input_cols_raw); in GetInputSizeInMklOrder() local
356 int input_planes, input_rows, input_cols; variable
|
| /aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/ |
| H A D | depthwise_depthfirst_multiplier.hpp | 305 auto input_cols = arm_gemm::roundup<size_t>(args.strategy->get_input_cols(), 16 / sizeof(T)); in sizeof_input_padding() local
321 auto input_cols = arm_gemm::roundup<size_t>(args.strategy->get_input_cols(), 16 / sizeof(T)); in sizeof_input_patch() local
|
| /aosp_15_r20/prebuilts/vndk/v30/arm64/include/external/libjpeg-turbo/simd/powerpc/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/arm/include/external/libjpeg-turbo/simd/loongson/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/x86/include/external/libjpeg-turbo/simd/loongson/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/arm/include/external/libjpeg-turbo/simd/powerpc/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/x86_64/include/external/libjpeg-turbo/simd/loongson/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/x86_64/include/external/libjpeg-turbo/simd/powerpc/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/x86/include/external/libjpeg-turbo/simd/powerpc/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|
| /aosp_15_r20/prebuilts/vndk/v30/arm64/include/external/libjpeg-turbo/simd/loongson/ |
| D | jcsample.h | 11 expand_right_edge(JSAMPARRAY image_data, int num_rows, JDIMENSION input_cols, in expand_right_edge()
|