Home
last modified time | relevance | path

Searched refs:row_size (Results 1 – 25 of 31) sorted by relevance

12

/external/tensorflow/tensorflow/lite/kernels/
Dmatrix_set_diag.cc65 const int row_size, const int col_size) { in FillDiagImpl() argument
68 for (int i = 0; i < row_size; i++) { in FillDiagImpl()
79 out += row_size * col_size; in FillDiagImpl()
80 in += row_size * col_size; in FillDiagImpl()
86 TfLiteTensor* output, const int batch_size, const int row_size, in FillDiag() argument
89 GetTensorData<T>(output), batch_size, row_size, col_size); in FillDiag()
102 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper() local
106 return FillDiag<int64_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
110 return FillDiag<int32_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
114 return FillDiag<int16_t>(input, diag, output, batch_size, row_size, in FillDiagHelper()
[all …]
Dmatrix_diag.cc64 void FillDiagImpl(const T* in, T* out, const int batch_size, const int row_size, in FillDiagImpl() argument
68 for (int i = 0; i < row_size; i++) { in FillDiagImpl()
79 out += row_size * col_size; in FillDiagImpl()
85 const int batch_size, const int row_size, const int col_size) { in FillDiag() argument
87 row_size, col_size); in FillDiag()
98 const int row_size = output->dims->data[num_output_dims - 2]; in FillDiagHelper() local
102 return FillDiag<int64_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
105 return FillDiag<int32_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
108 return FillDiag<int16_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
111 return FillDiag<int8_t>(input, output, batch_size, row_size, col_size); in FillDiagHelper()
[all …]
Dtopk_v2.cc96 TopContainer(int32 k, int32 row_size) : k_(k) { in TopContainer() argument
97 container_.reserve(std::min(k, row_size) + 1); in TopContainer()
176 void TopK(int32 row_size, int32 num_rows, const T* data, int32 k, in TopK() argument
178 TopContainer<T> topc(k, row_size); in TopK()
180 const T* values_row = data + row * row_size; in TopK()
182 for (int32 c = 0; c < row_size; ++c) { in TopK()
248 const int32 row_size = input->dims->data[input->dims->size - 1]; in Eval() local
255 TopK(row_size, num_rows, GetTensorData<float>(input), k, in Eval()
259 TopK(row_size, num_rows, input->data.uint8, k, output_indexes->data.i32, in Eval()
263 TopK(row_size, num_rows, input->data.int8, k, output_indexes->data.i32, in Eval()
[all …]
Dembedding_lookup.cc73 const int row_size = SizeOfDimension(value, 0); in EvalSimple() local
74 const int row_bytes = value->bytes / row_size; in EvalSimple()
81 if (idx >= row_size || idx < 0) { in EvalSimple()
85 idx, row_size - 1); in EvalSimple()
99 const int row_size = SizeOfDimension(value, 0); in EvalHybrid() local
114 if (idx >= row_size || idx < 0) { in EvalHybrid()
118 idx, row_size - 1); in EvalHybrid()
/external/pdfium/core/fxcodec/flate/
Dflatemodule.cpp298 const uint32_t row_size = CalculatePitch8(bpc, nColors, nPixels).ValueOrDie(); in PNG_PredictLine() local
302 memmove(pDestData, pSrcData + 1, row_size); in PNG_PredictLine()
305 for (uint32_t byte = 0; byte < row_size; ++byte) { in PNG_PredictLine()
366 const int row_size = (Colors * BitsPerComponent * Columns + 7) / 8; in PNG_Predictor() local
367 if (row_size <= 0) in PNG_Predictor()
369 const int row_count = (*data_size + row_size) / (row_size + 1); in PNG_Predictor()
372 const int last_row_size = *data_size % (row_size + 1); in PNG_Predictor()
374 FX_Alloc2D(uint8_t, row_size, row_count)); in PNG_Predictor()
382 int move_size = row_size; in PNG_Predictor()
392 for (int byte = 0; byte < row_size && byte_cnt < *data_size; in PNG_Predictor()
[all …]
/external/tensorflow/tensorflow/lite/examples/label_image/
Dbitmap_helpers.cc31 std::vector<uint8_t> decode_bmp(const uint8_t* input, int row_size, int width, in decode_bmp() argument
40 src_pos = ((height - 1 - i) * row_size) + j * channels; in decode_bmp()
42 src_pos = i * row_size + j * channels; in decode_bmp()
107 const int row_size = (8 * *channels * *width + 31) / 32 * 4; in read_bmp() local
115 return decode_bmp(bmp_pixels, row_size, *width, abs(*height), *channels, in read_bmp()
/external/tensorflow/tensorflow/core/kernels/
Dconcat_lib_cpu.h42 int64 row_size = 0; in ConcatCPUImpl() local
45 row_size += sizes.back(); in ConcatCPUImpl()
79 auto work = [&row_size, &sizes, &inputs, &output, &copier, &num_inputs]( in ConcatCPUImpl()
81 int64 skipped_rows = start / row_size; in ConcatCPUImpl()
82 T* out = output->data() + skipped_rows * row_size; in ConcatCPUImpl()
Dreverse_op.cc51 const int64 row_size = inner_size * middle_size; in ReverseRows() local
57 in_ptr += start * row_size; in ReverseRows()
58 out_ptr += start * row_size; in ReverseRows()
61 out_ptr += row_size; in ReverseRows()
70 out_ptr += row_size; in ReverseRows()
Dragged_range_op.cc93 SPLITS_TYPE row_size = rt_nested_splits(row + 1) - rt_nested_splits(row); in Compute() local
96 for (SPLITS_TYPE i = 0; i < row_size; ++i) { in Compute()
Dslice_op.cc178 const int64 row_size = size[0]; in Compute() local
183 for (int i = 0; i < row_size; ++i) { in Compute()
/external/crosvm/gpu_display/src/
Dgpu_display_wl.rs73 row_size: u32, field
206 let row_size = width * BYTES_PER_PIXEL; in create_surface() localVariable
207 let fb_size = row_size * height; in create_surface()
227 row_size, in create_surface()
240 row_size, in create_surface()
264 surface.row_size, in framebuffer()
/external/oboe/samples/RhythmGame/third_party/glm/detail/
Dtype_gentype.inl17 typename base<vT, cT, rT, pT>::size_type base<vT, cT, rT, pT>::row_size() function in glm::detail::base
193 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
209 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
225 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
241 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
257 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
273 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
289 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
305 typename base<vT, cT, rT, pT>::size_type stop_row = x.row_size();
318 typename base<vT, cT, rT, pT>::size_type stop_row = row_size();
[all …]
/external/mesa3d/src/gallium/auxiliary/util/
Du_framebuffer.c268 unsigned row_size = grid_width * samples; in util_sample_locations_flip_y() local
269 for (i = 0; i < row_size; i++) { in util_sample_locations_flip_y()
273 new_locations[dest_row * row_size + i] = locations[row * row_size + i]; in util_sample_locations_flip_y()
/external/tensorflow/tensorflow/core/kernels/image/
Ddecode_image_op.cc589 const int row_size = (img_channels * width + 3) / 4 * 4; in DecodeBmpV2() local
593 int size_diff = input.size() - header_size - (row_size * abs_height); in DecodeBmpV2()
602 (abs_height - 1) * row_size + in DecodeBmpV2()
628 DecodeBMP(bmp_pixels, row_size, output->flat<uint8>().data(), width, in DecodeBmpV2()
633 DecodeBMP(bmp_pixels, row_size, buffer.get(), width, abs_height, in DecodeBmpV2()
654 void DecodeBMP(const uint8* input, const int row_size, uint8* const output,
674 void DecodeImageV2Op::DecodeBMP(const uint8* input, const int row_size, in DecodeBMP() argument
684 src_pos = ((height - 1 - i) * row_size) + j * input_channels; in DecodeBMP()
686 src_pos = i * row_size + j * input_channels; in DecodeBMP()
Dscale_and_translate_op_test.cc216 const int64 row_size = num_col_squares * square_size * num_channels; in SetCheckerboardImageInput() local
217 const int64 image_size = num_row_squares * square_size * row_size; in SetCheckerboardImageInput()
229 auto it = data.begin() + b * image_size + r * row_size + in SetCheckerboardImageInput()
/external/mesa3d/src/amd/addrlib/src/chip/r800/
Dsi_gb_reg.h68 unsigned int row_size : 2; member
78 unsigned int row_size : 2; member
/external/mesa3d/src/mesa/main/
Dget_hash_generator.py89 row_size = 4
90 for i in range(0, hash_table_size, row_size):
91 row = dense_table[i : i + row_size]
/external/deqp/external/openglcts/modules/glesext/geometry_shader/
DesextcGeometryShaderClipping.cpp165 const unsigned int row_size = m_texture_width * m_texture_n_components; in iterate() local
225 unsigned char* data_row = buffer + y * row_size; in iterate()
DesextcGeometryShaderLayeredFramebuffer.cpp719 const int row_size = TEXTURE_WIDTH * pixel_size; in iterate() local
720 unsigned char* row = (unsigned char*)buffer + y * row_size; in iterate()
1396 int row_size = pixel_size * TEXTURE_WIDTH; in iterate() local
1397 unsigned char* ref_row = ref_data + y * row_size; in iterate()
1398 unsigned char* result_row = result_data + y * row_size; in iterate()
/external/libdrm/radeon/
Dradeon_surface.c106 uint32_t row_size; member
554 surf_man->hw_info.row_size = 1024; in eg_init_hw_info()
557 surf_man->hw_info.row_size = 2048; in eg_init_hw_info()
560 surf_man->hw_info.row_size = 4096; in eg_init_hw_info()
563 surf_man->hw_info.row_size = 4096; in eg_init_hw_info()
974 surf->tile_split = surf_man->hw_info.row_size; in eg_surface_best()
975 surf->stencil_tile_split = surf_man->hw_info.row_size / 2; in eg_surface_best()
1270 surf_man->hw_info.row_size = 1024; in si_init_hw_info()
1273 surf_man->hw_info.row_size = 2048; in si_init_hw_info()
1276 surf_man->hw_info.row_size = 4096; in si_init_hw_info()
[all …]
/external/ImageMagick/coders/
Dpsd.c1080 row_size; in ReadPSDChannelRaw() local
1093 row_size=GetPSDRowSize(image); in ReadPSDChannelRaw()
1094 pixels=(unsigned char *) AcquireQuantumMemory(row_size,sizeof(*pixels)); in ReadPSDChannelRaw()
1098 (void) memset(pixels,0,row_size*sizeof(*pixels)); in ReadPSDChannelRaw()
1105 count=ReadBlob(image,row_size,pixels); in ReadPSDChannelRaw()
1106 if (count != (ssize_t) row_size) in ReadPSDChannelRaw()
1149 row_size; in ReadPSDChannelRLE() local
1163 row_size=GetPSDRowSize(image); in ReadPSDChannelRLE()
1164 pixels=(unsigned char *) AcquireQuantumMemory(row_size,sizeof(*pixels)); in ReadPSDChannelRLE()
1174 if (length > (row_size+2048)) /* arbitrary number */ in ReadPSDChannelRLE()
[all …]
/external/libaom/libaom/av1/encoder/x86/
Dav1_txfm1d_sse4.h81 const int row_size = txfm_size; in transpose_32() local
86 for (r = 0; r < row_size; r += 4) { in transpose_32()
/external/tflite-support/tensorflow_lite_support/custom_ops/kernel/ragged/
Dragged_range_tflite.cc122 const SPLITS_TYPE row_size = in EvalT() local
126 for (SPLITS_TYPE i = 0; i < row_size; ++i) { in EvalT()
/external/mesa3d/src/mesa/drivers/dri/i915/
Dintel_mipmap_tree.c566 int row_size = cpp * width; in intel_miptree_copy_slice_sw() local
567 if (src_stride == row_size && in intel_miptree_copy_slice_sw()
568 dst_stride == row_size) { in intel_miptree_copy_slice_sw()
569 memcpy(dst, src, row_size * height); in intel_miptree_copy_slice_sw()
572 memcpy(dst, src, row_size); in intel_miptree_copy_slice_sw()
/external/deqp/external/openglcts/modules/glesext/tessellation_shader/
DesextcTessellationShaderPoints.cpp566 const unsigned int row_size = pixel_size * m_rt_width; in iterate() local
579 const unsigned char* rendered_color_ubyte_ptr = buffer + row_size * y + x * pixel_size; in iterate()

12