Home
last modified time | relevance | path

Searched defs:input_rows (Results 1 – 25 of 35) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Deigen_backward_spatial_convolutions_test.cc32 const int input_rows = 3; in TEST() local
81 const int input_rows = 3; in TEST() local
133 const int input_rows = 7; in TEST() local
181 const int input_rows = 7; in TEST() local
233 const int input_rows = 7; in TEST() local
288 const int input_rows = 7; in TEST() local
341 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid()
400 int input_rows = 3; in TEST() local
417 const int num_batches, const int input_depth, const int input_rows, in test_batched_strided_spatial_convolution_backward_input_valid_row_major()
477 int input_rows = 3; in TEST() local
[all …]
H A Deigen_pooling_test.cc29 const int input_rows = 5; in TEST() local
77 const int input_rows = 5; in TEST() local
127 const int input_rows = 5; in TEST() local
187 const int input_rows = 5; in TEST() local
247 const int input_rows = 5; in TEST() local
309 const int input_rows = 5; in TEST() local
371 const int input_rows = 5; in TEST() local
449 const int input_rows = 5; in TEST() local
526 const int input_rows = 5; in TEST() local
574 const int input_rows = 5; in TEST() local
[all …]
H A Deigen_backward_cuboid_convolutions_test.cc33 const int input_rows = 3; in TEST() local
96 const int input_rows = 3; in TEST() local
159 const int input_rows = 3; in TEST() local
230 const int input_rows = 3; in TEST() local
302 const int input_rows = 3; in TEST() local
369 const int input_rows = 3; in TEST() local
435 const int input_rows = 3; in TEST() local
501 const int input_rows = 3; in TEST() local
568 const int input_rows = 7; in TEST() local
635 const int input_rows = 7; in TEST() local
[all …]
H A Ddilation_ops_gpu.cu.cc40 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()
186 const int input_rows = input.dimension(1); in operator ()() local
217 const int input_rows = input.dimension(1); in operator ()() local
259 const int input_rows = input.dimension(1); in operator ()() local
H A Deigen_spatial_convolutions_test.cc31 const int input_rows = 4; in TEST() local
76 const int input_rows = 4; in TEST() local
195 const int input_rows = 5; in TEST() local
247 const int input_rows = 5; in TEST() local
305 const int input_rows = 5; in TEST() local
361 const int input_rows = 5; in TEST() local
409 const int input_rows = 3; in TEST() local
457 const int input_rows = 5; in TEST() local
508 const int input_rows = 7; in TEST() local
559 const int input_rows = 7; in TEST() local
[all …]
H A Ddilation_ops.cc76 const int input_rows = input.dim_size(1); in ParseSizes() local
171 const int input_rows = input.dimension(1); in operator ()() local
286 const int input_rows = input.dimension(1); in operator ()() local
414 const int input_rows = input.dimension(1); in operator ()() local
H A Dconv_ops.cc367 const Tensor& filter, int batch, int input_rows, in Run() argument
383 const Tensor& filter, int batch, int input_rows, in Run()
424 const Tensor& filter, int batch, int input_rows, in Run()
438 const Tensor& filter, int batch, int input_rows, in Run()
589 const int input_rows = static_cast<int>(input_rows_raw); in ComputeConv2DDimension() local
H A Deigen_benchmark.h87 Eigen::Index input_rows = input_dims[1]; in SpatialConvolutionBackwardInput() local
220 Eigen::Index input_rows = input_dims[1]; in CuboidConvolutionBackwardInput() local
/aosp_15_r20/external/libopus/src/
H A Dmapping_matrix.c88 int input_rows, in mapping_matrix_multiply_channel_in_float()
123 int input_rows, in mapping_matrix_multiply_channel_out_float()
159 int input_rows, in mapping_matrix_multiply_channel_in_short()
200 int input_rows, in mapping_matrix_multiply_channel_out_short()
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/convolution/winograd/
H A Dinput_transform.hpp145 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 Dcxx11_tensor_image_patch.cpp186 int input_rows = 3; in test_patch_padding_valid() local
262 int input_rows = 5; in test_patch_padding_valid_same_value() local
335 int input_rows = 4; in test_patch_padding_same() local
412 int input_rows = 15; in test_patch_padding_same_negative_padding_clip_to_zero() local
H A Dcxx11_tensor_image_patch_sycl.cpp261 IndexType input_rows = 3; in test_patch_padding_valid_sycl() local
367 IndexType input_rows = 5; in test_patch_padding_valid_same_value_sycl() local
464 IndexType input_rows = 4; in test_patch_padding_same_sycl() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Druntime_conv2d_acl.cc31 int64_t input_batch, int64_t input_rows, int64_t input_cols, in ACLDepthwiseConvImpl()
128 float* rhs, int64_t input_batch, int64_t input_rows, in ACLGemmConvImpl()
235 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_ACLConv2DF32()
H A Druntime_single_threaded_conv2d.cc24 Eigen::half* rhs, int64_t input_batch, int64_t input_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 Druntime_conv2d.cc27 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_EigenConv2DF32()
49 Eigen::half* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_EigenConv2DF16()
H A Druntime_conv2d_mkl.cc53 ScalarType* rhs, int64_t input_batch, int64_t input_rows, in MKLConvImpl()
156 int64_t input_batch, int64_t input_rows, int64_t input_cols, in __xla_cpu_runtime_MKLConv2DF32()
H A Druntime_conv2d_acl.h79 float* lhs, float* rhs, int64_t input_batch, int64_t input_rows, in __xla_cpu_runtime_ACLConv2DF32()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cpu/
H A Dqembeddingbag_unpack.cpp27 const auto input_rows = packed_weight.size(0); in unpack() local
123 const int64_t input_rows = c10::size_to_dim_(col_dim, packed_weight_sizes); in qembeddingbag_byte_unpack_out() local
188 const auto input_rows = packed_weight.size(0); in _qembeddingbag_nbit_unpack_helper() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_conv_ops.h136 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder() local
159 int input_rows = static_cast<int>(input_rows_raw); in GetInputSizeInMklOrder() local
356 int input_planes, input_rows, input_cols; variable
/aosp_15_r20/external/pytorch/aten/src/ATen/native/quantized/cuda/
H A DEmbeddingBag.cu175 const int32_t input_rows = c10::size_to_dim_(col_dim, packed_weight_sizes); in qembeddingbag_byte_unpack() local
546 const auto input_rows = packed_weight.size(0); in qembeddingbag_4bit_unpack() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/arm_conv/depthwise/kernels/
H A Dsve_fp32_nhwc_3x3_s1_output2x2_mla_depthfirst_strided.hpp55 constexpr static unsigned int input_rows = 4; member
/aosp_15_r20/external/libjpeg-turbo/
H A Djcprepct.c109 expand_bottom_edge(JSAMPARRAY image_data, JDIMENSION num_cols, int input_rows, in expand_bottom_edge()
/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
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/assembly/
H A Ddepthwise.hpp52 unsigned int n_batches, input_rows, input_cols, input_channels; member
H A Dpooling.hpp53 unsigned int n_batches, input_rows, input_cols, n_channels; member

12