Home
last modified time | relevance | path

Searched defs:row_size (Results 1 – 25 of 49) sorted by relevance

12

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DScanUtils.cuh20 constexpr inline integer get_log_num_threads_x_inner_scan(integer num_rows, integer row_size) { in get_log_num_threads_x_inner_scan()
62 int num_rows, int row_size, in tensor_kernel_scan_innermost_dim_with_indices()
156 …const uint32_t num_orows, const uint32_t num_irows, const uint32_t row_size, scalar_t init, Binary… in tensor_kernel_scan_outer_dim_with_indices()
192 int64_t row_size = self.size(dim); in scan_outer_dim_with_indices() local
222 int row_size = self.size(ndim - 1); in scan_innermost_dim_with_indices() local
273 … const uint32_t num_orows, const uint32_t num_irows, const uint32_t row_size, in tensor_kernel_scan_outer_dim()
305 const uint32_t num_rows, const uint32_t row_size, in tensor_kernel_scan_innermost_dim_impl()
376 const uint32_t row_size, in tensor_kernel_scan_innermost_dim()
393 const int64_t row_size = self.size(dim); in scan_outer_dim() local
421 int64_t row_size = self.size(ndim - 1); in scan_innermost_dim() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmatrix_diag.cc64 void FillDiagImpl(const T* in, T* out, const int batch_size, const int row_size, in FillDiagImpl()
85 const int batch_size, const int row_size, const int col_size) { in FillDiag()
98 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper() local
H A Dmatrix_set_diag.cc65 const int row_size, const int col_size) { in FillDiagImpl()
86 TfLiteTensor* output, const int batch_size, const int row_size, in FillDiag()
102 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper() local
H A Dtopk_v2.cc96 TopContainer(int32 k, int32 row_size) : k_(k) { in TopContainer()
177 void TopK(int32 row_size, int32 num_rows, const T* data, int32 k, in TopK()
250 const int32 row_size = input->dims->data[input->dims->size - 1]; in Eval() local
H A Dembedding_lookup.cc73 const int row_size = SizeOfDimension(value, 0); in EvalSimple() local
103 const int row_size = SizeOfDimension(value, 0); in EvalHybrid() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/
H A Dapprox_topk_test.py71 def test_non_fused_max_k(self, k, row_size, num_rows, aggregate_to_topk): argument
94 def test_non_fused_min_k(self, k, row_size, num_rows, aggregate_to_topk): argument
193 def test_gradients(self, dtype, k, row_size, num_rows): argument
/aosp_15_r20/external/tensorflow/tensorflow/python/kernel_tests/math_ops/
H A Dapprox_topk_test.py73 def test_non_fused_max_k(self, dtype, k, row_size, num_rows, argument
96 def test_non_fused_min_k(self, dtype, k, row_size, num_rows, argument
189 def test_gradients(self, dtype, k, row_size, num_rows): argument
/aosp_15_r20/external/ComputeLibrary/utils/
H A DImageLoader.h80 void get_row(uint8_t *dst, size_t row_size) override in get_row()
114 void get_row(uint8_t *dst, size_t row_size) override in get_row()
227 size_t row_size = _width * image.info()->element_size(); in fill_image() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/examples/label_image/
H A Dbitmap_helpers.cc32 std::vector<uint8_t> decode_bmp(const uint8_t* input, int row_size, int width, in decode_bmp()
108 const int row_size = (8 * *channels * *width + 31) / 32 * 4; in read_bmp() local
/aosp_15_r20/external/mesa3d/src/amd/addrlib/src/chip/r800/
H A Dsi_gb_reg.h50 unsigned int row_size : 2; member
60 unsigned int row_size : 2; member
94 unsigned int row_size : 2; member
104 unsigned int row_size : 2; member
/aosp_15_r20/external/ComputeLibrary/tests/validation/UNIT/
H A DWindowIterator.cpp75 int row_size = 0; variable
114 int row_size = 0; variable
/aosp_15_r20/external/pdfium/core/fxcodec/flate/
H A Dflatemodule.cpp316 const uint32_t row_size = fxge::CalculatePitch8OrDie(bpc, nColors, nPixels); in PNG_PredictLine() local
384 const int row_size = (Colors * BitsPerComponent * Columns + 7) / 8; in PNG_Predictor() local
473 uint32_t row_size, in TIFF_PredictLine()
517 int row_size = (Colors * BitsPerComponent * Columns + 7) / 8; in TIFF_Predictor() local
/aosp_15_r20/external/libyuv/source/
H A Dscale_argb.cc168 const int row_size = (dst_width * 2 * 4 + 31) & ~31; in ScaleARGBDown4Box() local
595 const int row_size = (dst_width * 4 + 31) & ~31; in ScaleARGBBilinearUp() local
867 const int row_size = (dst_width * 4 + 31) & ~31; in ScaleYUVToARGBBilinearUp() local
H A Dconvert_argb.cc6245 const int row_size = (width + 31) & ~31; in I420ToARGBMatrixBilinear() local
6383 const int row_size = (width + 31) & ~31; in I422ToARGBMatrixLinear() local
6516 const int row_size = (width + 31) & ~31; in I420ToRGB24MatrixBilinear() local
6624 const int row_size = (width + 31) & ~31; in I010ToAR30MatrixBilinear() local
6725 const int row_size = (width + 31) & ~31; in I210ToAR30MatrixLinear() local
6815 const int row_size = (width + 31) & ~31; in I010ToARGBMatrixBilinear() local
6915 const int row_size = (width + 31) & ~31; in I210ToARGBMatrixLinear() local
7093 const int row_size = (width + 31) & ~31; in I420AlphaToARGBMatrixBilinear() local
7295 const int row_size = (width + 31) & ~31; in I422AlphaToARGBMatrixLinear() local
7436 const int row_size = (width + 31) & ~31; in I010AlphaToARGBMatrixBilinear() local
[all …]
H A Dconvert.cc2697 const int row_size = (width * 4 + 31) & ~31; in RGB24ToI420() local
2875 const int row_size = (width * 4 + 31) & ~31; in RGB24ToJ420() local
3056 const int row_size = (width * 4 + 31) & ~31; in RAWToI420() local
3234 const int row_size = (width * 4 + 31) & ~31; in RAWToJ420() local
3414 const int row_size = (width * 4 + 31) & ~31; in RGB565ToI420() local
3596 const int row_size = (width * 4 + 31) & ~31; in ARGB1555ToI420() local
3811 const int row_size = (width * 4 + 31) & ~31; in ARGB4444ToI420() local
H A Dscale_uv.cc205 const int row_size = (dst_width * 2 * 2 + 15) & ~15; in ScaleUVDown4Box() local
612 const int row_size = (dst_width * 2 + 15) & ~15; in ScaleUVBilinearUp() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Ddecode_image_op.cc599 const int row_size = (img_channels * width + 3) / 4 * 4; in DecodeBmpV2() local
684 void DecodeImageV2Op::DecodeBMP(const uint8* input, const int row_size, in DecodeBMP()
/aosp_15_r20/external/crosvm/gpu_display/src/
H A Dgpu_display_wl.rs107 row_size: u32, field
389 let row_size = width * BYTES_PER_PIXEL; in create_surface() localVariable
/aosp_15_r20/external/deqp/external/openglcts/modules/glesext/geometry_shader/
H A DesextcGeometryShaderLayeredFramebuffer.cpp721 const int row_size = TEXTURE_WIDTH * pixel_size; in iterate() local
1398 int row_size = pixel_size * TEXTURE_WIDTH; in iterate() local
H A DesextcGeometryShaderClipping.cpp165 const unsigned int row_size = m_texture_width * m_texture_n_components; in iterate() local
/aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/
DEmbeddingLookup.cpp37 const int row_size = value_->shape().dimensions[0]; in Eval() local
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dragged_range_op.cc111 SPLITS_TYPE row_size = rt_nested_splits(row + 1) - rt_nested_splits(row); in Compute() local
H A Dconcat_lib_cpu.h42 int64_t row_size = 0; in ConcatCPUImpl() local
/aosp_15_r20/external/mesa3d/src/gallium/auxiliary/util/
H A Du_framebuffer.c280 unsigned row_size = grid_width * samples; in util_sample_locations_flip_y() local
/aosp_15_r20/external/libaom/av1/encoder/x86/
H A Dav1_txfm1d_sse4.h81 const int row_size = txfm_size; in transpose_32() local

12