Home
last modified time | relevance | path

Searched defs:output_rows (Results 1 – 25 of 87) sorted by relevance

1234

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc37 const int output_rows = input_rows - patch_rows + 1; in TEST() local
86 const int output_rows = input_rows - patch_rows + 1; in TEST() local
138 const int output_rows = input_rows; in TEST() local
186 const int output_rows = input_rows; in TEST() local
238 const int output_rows = input_rows - patch_rows + 1; in TEST() local
293 const int output_rows = input_rows - patch_rows + 1; in TEST() local
348 const int output_rows = divup(input_rows - patch_rows + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid() local
424 const int output_rows = divup(input_rows - patch_rows + 1, stride); in test_batched_strided_spatial_convolution_backward_input_valid_row_major() local
501 const int output_rows = input_rows - patch_rows + 1; in TEST() local
555 const int output_rows = input_rows - patch_rows + 1; in TEST() local
[all …]
H A Deigen_pooling_test.cc34 const int output_rows = 2; in TEST() local
82 const int output_rows = 2; in TEST() local
133 const int output_rows = 2; in TEST() local
193 const int output_rows = 2; in TEST() local
253 const int output_rows = 2; in TEST() local
315 const int output_rows = 2; in TEST() local
377 const int output_rows = input_rows; in TEST() local
455 const int output_rows = input_rows; in TEST() local
531 const int output_rows = 2; in TEST() local
579 const int output_rows = 2; in TEST() local
[all …]
H A Deigen_backward_cuboid_convolutions_test.cc38 const int output_rows = input_rows - patch_rows + 1; in TEST() local
101 const int output_rows = input_rows - patch_rows + 1; in TEST() local
164 const int output_rows = input_rows; in TEST() local
235 const int output_rows = input_rows; in TEST() local
307 const int output_rows = input_rows - patch_rows + 1; in TEST() local
374 const int output_rows = input_rows - patch_rows + 1; in TEST() local
441 const int output_rows = input_rows - patch_rows + 1; in TEST() local
507 const int output_rows = input_rows - patch_rows + 1; in TEST() local
574 const int output_rows = input_rows - patch_rows + 1; in TEST() local
641 const int output_rows = input_rows - patch_rows + 1; in TEST() local
[all …]
H A Ddilation_ops_gpu.cu.cc41 int depth, int filter_rows, int filter_cols, int output_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()
193 const int output_rows = output.dimension(1); in operator ()() local
224 const int output_rows = out_backprop.dimension(1); in operator ()() local
266 const int output_rows = out_backprop.dimension(1); in operator ()() local
H A Deigen_spatial_convolutions_test.cc36 const int output_rows = input_rows; in TEST() local
81 const int output_rows = input_rows; in TEST() local
201 const int output_rows = input_rows - patch_rows + 1; in TEST() local
256 const int output_rows = 2; in TEST() local
311 const int output_rows = input_rows - patch_rows + 1; in TEST() local
367 const int output_rows = 2; in TEST() local
415 const int output_rows = 2; in TEST() local
463 const int output_rows = 2; in TEST() local
514 const int output_rows = 3; in TEST() local
565 const int output_rows = 3; in TEST() local
[all …]
H A Dsegment_reduction_ops_impl.h104 const Index output_rows = in Compute() local
285 Index output_rows = *output_rows_host.data(); in ComputeAsync() local
H A Ddilation_ops.cc178 const int output_rows = output.dimension(1); in operator ()() local
293 const int output_rows = out_backprop.dimension(1); in operator ()() local
421 const int output_rows = out_backprop.dimension(1); in operator ()() local
/aosp_15_r20/external/libopus/src/
H A Dmapping_matrix.c91 int output_rows, in mapping_matrix_multiply_channel_in_float()
125 int output_rows, in mapping_matrix_multiply_channel_out_float()
162 int output_rows, in mapping_matrix_multiply_channel_in_short()
202 int output_rows, in mapping_matrix_multiply_channel_out_short()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc33 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in ACLDepthwiseConvImpl()
132 int64_t output_rows, int64_t output_cols, in ACLGemmConvImpl()
237 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, 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()
46 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_EigenSingleThreadedConv2DF32()
H A Druntime_conv2d.cc29 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_EigenConv2DF32()
52 int64_t output_rows, int64_t output_cols, int64_t row_stride, in __xla_cpu_runtime_EigenConv2DF16()
H A Druntime_conv2d_mkl.cc57 int64_t output_rows, int64_t output_cols, int64_t row_stride, in MKLConvImpl()
158 int64_t kernel_channels, int64_t kernel_filters, int64_t output_rows, in __xla_cpu_runtime_MKLConv2DF32()
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dcommon_shape_fns.cc120 DimensionHandle output_rows = transpose_a ? c->Dim(a, 1) : c->Dim(a, 0); in MatMulShape() local
320 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); in BatchMatMulV2Shape() local
357 DimensionHandle output_rows = c->Dim(a_shape, adj_x ? -1 : -2); 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/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
H A Dfully_connected.h57 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnectedPerChannel() local
129 const int output_rows = output_shape.Dims(output_dim_count - 1); in FullyConnected() local
H A Ddepthwise_conv_hybrid.h67 const int output_rows = output_shape.Dims(1); in DepthwiseConvHybridGeneral() local
408 const int output_rows = output_shape.Dims(1); in DepthwiseConvHybridPerChannel() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/
H A Daddressing.hpp190 unsigned int output_rows, unsigned int output_cols, in fill_pointer_array_generic_kernel()
239 unsigned int output_rows, unsigned int output_cols, in fill_patch_array_generic_kernel()
H A Daddressing.cpp91 const unsigned int output_rows, const unsigned int output_cols, in fill_pointer_array_generic_kernel()
262 const unsigned int output_rows, const unsigned int output_cols, in fill_patch_array_generic_kernel()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/pooling/
H A Dpooling_depthfirst.hpp41 unsigned int input_rows, input_cols, output_rows, output_cols; member in arm_conv::pooling::DepthfirstStrategy
46 unsigned int output_rows, unsigned int output_cols) in DepthfirstStrategy()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
H A Doutput_transform.hpp150 unsigned int output_rows, unsigned int output_cols, in TransformBase()
272 unsigned int output_rows, unsigned int output_cols, in TransformUnpadded()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Dsparse_csr_matrix_ops.cc269 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anona88575120702() local
426 auto output_rows = c->Dim(a_shape, transpose_a ? -1 : -2); in __anona88575120a02() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/
H A Ddepthwise_depthfirst_multiplier.hpp132 …GenericDepthfirstMultiplierKernelStrategy(unsigned int output_rows, unsigned int output_cols, arm_… in GenericDepthfirstMultiplierKernelStrategy()
161 …GenericDepthfirstMultiplierKernelStrategy(unsigned int output_rows, unsigned int output_cols, arm_… in GenericDepthfirstMultiplierKernelStrategy()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/
H A Dsme2_s8q_planar_3x3_s1_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s1_4rows_dot_za
H A Dsme2_s8q_planar_3x3_s2_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_s8q_planar_3x3_s2_4rows_dot_za
H A Dsme2_s8q_planar_5x5_s2_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s2_4rows_dot_za
H A Dsme2_u8s8u8q_planar_3x3_s2_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_3x3_s2_4rows_dot_za

1234