Home
last modified time | relevance | path

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

1234

/external/tensorflow/tensorflow/core/kernels/
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 …]
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 …]
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
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 …]
Dsegment_reduction_ops_impl.h104 const Index output_rows = in Compute() local
285 Index output_rows = *output_rows_host.data(); in ComputeAsync() local
/external/libopus/src/
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()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
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()
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()
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()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/integer_ops/
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
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/
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()
/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/
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
Dsme2_u8q_planar_3x3_s1_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8q_planar_3x3_s1_4rows_dot_za
Dsme2_u8q_planar_3x3_s2_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8q_planar_3x3_s2_4rows_dot_za
Dsme2_u8s8u8q_planar_5x5_s1_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s1_4rows_dot_za
Dsme2_fp32_planar_3x3_s2_4rows_mla_za.hpp55 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s2_4rows_mla_za
Dsme2_u8s8u8q_planar_5x5_s2_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_5x5_s2_4rows_dot_za
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
Dsme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za.hpp55 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_3x3_s1_4rows_dot_za
Dsme2_s8q_planar_5x5_s1_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_s8q_planar_5x5_s1_4rows_dot_za
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
Dsme2_u8s8u8q_planar_3x3_s1_4rows_dot_za.hpp53 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_u8s8u8q_planar_3x3_s1_4rows_dot_za
Dsme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za.hpp55 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_fp32bf16fp32_planar_5x5_s2_4rows_dot_za
Dsme2_fp32_planar_3x3_s1_4rows_mla_za.hpp55 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_fp32_planar_3x3_s1_4rows_mla_za
Dsme2_fp32_planar_5x5_s1_4rows_mla_za.hpp55 constexpr static auto output_rows = 4u; member in arm_conv::depthwise::sme2_fp32_planar_5x5_s1_4rows_mla_za

1234