Home
last modified time | relevance | path

Searched defs:output_cols (Results 1 – 25 of 86) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc38 const int output_cols = input_cols - patch_cols + 1; in TEST() local
87 const int output_cols = input_cols - patch_cols + 1; in TEST() local
139 const int output_cols = input_cols; in TEST() local
187 const int output_cols = input_cols; in TEST() local
239 const int output_cols = input_cols - patch_cols + 1; in TEST() local
294 const int output_cols = input_cols - patch_cols + 1; in TEST() local
349 const int output_cols = divup(input_cols - patch_cols + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid() local
425 const int output_cols = divup(input_cols - patch_cols + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid_row_major() local
502 const int output_cols = input_cols - patch_cols + 1; in TEST() local
556 const int output_cols = input_cols - patch_cols + 1; in TEST() local
[all …]
H A Deigen_pooling_test.cc35 const int output_cols = 2; in TEST() local
83 const int output_cols = 2; in TEST() local
134 const int output_cols = 3; in TEST() local
194 const int output_cols = 3; in TEST() local
254 const int output_cols = 3; in TEST() local
316 const int output_cols = 3; in TEST() local
378 const int output_cols = input_cols; in TEST() local
456 const int output_cols = input_cols; in TEST() local
532 const int output_cols = 2; in TEST() local
580 const int output_cols = 2; in TEST() local
[all …]
H A Deigen_backward_cuboid_convolutions_test.cc39 const int output_cols = input_cols - patch_cols + 1; in TEST() local
102 const int output_cols = input_cols - patch_cols + 1; in TEST() local
165 const int output_cols = input_cols; in TEST() local
236 const int output_cols = input_cols; in TEST() local
308 const int output_cols = input_cols - patch_cols + 1; in TEST() local
375 const int output_cols = input_cols - patch_cols + 1; in TEST() local
442 const int output_cols = input_cols - patch_cols + 1; in TEST() local
508 const int output_cols = input_cols - patch_cols + 1; in TEST() local
575 const int output_cols = input_cols - patch_cols + 1; in TEST() local
642 const int output_cols = input_cols - patch_cols + 1; in TEST() local
[all …]
H A Ddilation_ops_gpu.cu.cc42 int output_cols, int stride_rows, int stride_cols, int rate_rows, in DilationKernel()
81 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropInputKernel()
131 int filter_cols, int output_rows, int output_cols, int stride_rows, in DilationBackpropFilterKernel()
194 const int output_cols = output.dimension(2); in operator ()() local
225 const int output_cols = out_backprop.dimension(2); in operator ()() local
267 const int output_cols = out_backprop.dimension(2); in operator ()() local
H A Deigen_spatial_convolutions_test.cc37 const int output_cols = input_cols; in TEST() local
82 const int output_cols = input_cols; in TEST() local
202 const int output_cols = input_cols - patch_cols + 1; in TEST() local
257 const int output_cols = 1; in TEST() local
312 const int output_cols = input_cols - patch_cols + 1; in TEST() local
368 const int output_cols = 2; in TEST() local
416 const int output_cols = 2; in TEST() local
464 const int output_cols = 2; in TEST() local
515 const int output_cols = 3; in TEST() local
566 const int output_cols = 3; in TEST() local
[all …]
/aosp_15_r20/external/libjpeg-turbo/
H A Djcsample.c95 JDIMENSION output_cols) in expand_right_edge()
152 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in int_downsample() local
224 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v1_downsample() local
260 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v2_downsample() local
302 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in h2v2_smooth_downsample() local
390 JDIMENSION output_cols = compptr->width_in_blocks * DCTSIZE; in fullsize_smooth_downsample() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc34 int64_t output_cols, int64_t row_stride, int64_t col_stride, in ACLDepthwiseConvImpl()
132 int64_t output_rows, int64_t output_cols, in ACLGemmConvImpl()
238 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_single_threaded_conv2d.cc27 int64_t output_rows, int64_t output_cols, int64_t row_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF16()
47 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
H A Druntime_conv2d.cc30 int64_t output_cols, int64_t row_stride, int64_t col_stride, in __xla_cpu_runtime_EigenConv2DF32()
52 int64_t output_rows, int64_t output_cols, int64_t row_stride, in __xla_cpu_runtime_EigenConv2DF16()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc121 DimensionHandle output_cols = transpose_b ? c->Dim(b, 0) : c->Dim(b, 1); in MatMulShape() local
321 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulV2Shape() local
358 DimensionHandle output_cols = c->Dim(b_shape, adj_y ? -2 : -1); in BatchMatMulShape() local
728 DimensionHandle output_rows, output_cols; in Conv2DShapeImpl() local
861 DimensionHandle output_planes, output_rows, output_cols; in Conv3DShape() local
1070 DimensionHandle output_rows, output_cols; in DepthwiseConv2DNativeShapeImpl() local
1164 DimensionHandle output_rows, output_cols; in AvgPoolShape() local
1654 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolShapeImpl() local
1779 DimensionHandle output_rows, output_cols, output_depth; in MaxPoolV2Shape() local
1856 DimensionHandle output_planes, output_rows, output_cols; in Pool3DShape() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8s8u8q_planar_3x3_s2_2rows_dot_za/
H A Dgeneric.cpp47 unsigned int output_cols, in sme2_u8s8u8q_planar_3x3_s2_2rows_dot_za_impl()
58 long unsigned int input_cols, output_cols; in sme2_u8s8u8q_planar_3x3_s2_2rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_s8q_planar_3x3_s2_2rows_dot_za/
H A Dgeneric.cpp47 unsigned int output_cols, in sme2_s8q_planar_3x3_s2_2rows_dot_za_impl()
58 long unsigned int input_cols, output_cols; in sme2_s8q_planar_3x3_s2_2rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8q_planar_3x3_s2_2rows_dot_za/
H A Dgeneric.cpp47 unsigned int output_cols, in sme2_u8q_planar_3x3_s2_2rows_dot_za_impl()
58 long unsigned int input_cols, output_cols; in sme2_u8q_planar_3x3_s2_2rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8q_planar_3x3_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8q_planar_3x3_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8q_planar_3x3_s1_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_s8q_planar_3x3_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_s8q_planar_3x3_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_s8q_planar_3x3_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8q_planar_3x3_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8q_planar_3x3_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8q_planar_3x3_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_s8q_planar_5x5_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_s8q_planar_5x5_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_s8q_planar_5x5_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8q_planar_5x5_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8q_planar_5x5_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8q_planar_5x5_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_s8q_planar_5x5_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_s8q_planar_5x5_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_s8q_planar_5x5_s1_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_s8q_planar_3x3_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_s8q_planar_3x3_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_s8q_planar_3x3_s1_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za_impl() member
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/sme2_u8q_planar_5x5_s1_4rows_dot_za/
H A Dgeneric.cpp49 unsigned int output_cols, in sme2_u8q_planar_5x5_s1_4rows_dot_za_impl()
61 long unsigned int input_cols, output_cols; in sme2_u8q_planar_5x5_s1_4rows_dot_za_impl() member

1234