Home
last modified time | relevance | path

Searched defs:num_rows (Results 1 – 25 of 135) sorted by relevance

123456

/external/tensorflow/tensorflow/core/kernels/linalg/
Dmatrix_diag_op_gpu.cu.cc31 const int num_rows, in ComputeContentOffset()
46 const int num_threads, const int num_rows, const int num_cols, in MatrixDiagKernel()
83 const int num_rows = output.dimension(1); in Compute() local
103 const int num_threads, const int num_rows, const int num_cols, in MatrixDiagPartKernel()
144 const int num_rows = input.dimension(1); in Compute() local
Dlu_op.cc59 double num_rows = static_cast<double>(input_matrix_shape.dim_size(0)); in GetCostPerUnit() local
84 const int64 num_rows = input.dim_size(input_rank - 2); in Compute() local
129 const Tensor& input, int64 num_rows, int64 num_cols, in ComputeTensorSlice()
Dlu_op_gpu.cu.cc40 int64 num_rows, const int* __restrict__ pivots, in ComputePermutationFromTranspositions()
65 GpuLaunchConfig config, const int64 num_rows, in ComputePermutationFromTranspositionsKernel()
94 const int64 num_rows = input.dim_size(input_rank - 2); in ComputeAsync() local
Dmatrix_diag_op.cc101 const Eigen::Index num_rows = input_shape.dim_size(rank - 2); in Compute() local
169 int32 num_rows = -1; in Compute() local
319 int diag_index, int max_diag_len, int num_rows, int num_cols, in ComputeDiagLenAndContentOffset()
342 const Eigen::Index num_rows = output.dimension(1); in Compute() local
401 const Eigen::Index num_rows = input.dimension(1); in Compute() local
Dmatrix_set_diag_op.cc103 const Eigen::Index num_rows = input_shape.dim_size(input_rank - 2); in Compute() local
221 const Eigen::Index num_rows = output.dimension(1); in Compute() local
/external/libjpeg-turbo/
Djdcolor.c255 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in ycc_rgb_convert()
327 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in rgb_gray_convert()
362 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in null_convert()
423 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in grayscale_convert()
436 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in gray_rgb_convert()
481 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in rgb_rgb_convert()
529 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in ycck_cmyk_convert()
657 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in ycc_rgb565_convert()
668 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in ycc_rgb565D_convert()
679 JDIMENSION input_row, JSAMPARRAY output_buf, int num_rows) in rgb_rgb565_convert()
[all …]
Djccolor.c235 JSAMPIMAGE output_buf, JDIMENSION output_row, int num_rows) in rgb_ycc_convert()
283 JSAMPIMAGE output_buf, JDIMENSION output_row, int num_rows) in rgb_gray_convert()
328 JSAMPIMAGE output_buf, JDIMENSION output_row, int num_rows) in rgb_rgb_convert()
377 JSAMPIMAGE output_buf, JDIMENSION output_row, int num_rows) in cmyk_ycck_convert()
428 JSAMPIMAGE output_buf, JDIMENSION output_row, int num_rows) in grayscale_convert()
456 JDIMENSION output_row, int num_rows) in null_convert()
Djdpostct.c138 JDIMENSION num_rows, max_rows; in post_process_1pass() local
170 JDIMENSION old_next_row, num_rows; in post_process_prepass() local
213 JDIMENSION num_rows, max_rows; in post_process_2pass() local
Djquant1.c464 JSAMPARRAY output_buf, int num_rows) in color_quantize()
492 JSAMPARRAY output_buf, int num_rows) in color_quantize3()
520 JSAMPARRAY output_buf, int num_rows) in quantize_ord_dither()
570 JSAMPARRAY output_buf, int num_rows) in quantize3_ord_dither()
612 JSAMPARRAY output_buf, int num_rows) in quantize_fs_dither()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dmatrix_diag_ops.cc30 static inline int ComputeDiagLen(int diag_index, int num_rows, int num_cols) { in ComputeDiagLen()
95 const int64 num_rows, in ValidateDiagIndexWithOutputMatrixSize()
121 const int64 num_rows, const int64 num_cols, in SetMatrixDiag()
258 int64 num_rows = -1; in Compile() local
382 const int64 num_rows = input_shape.dim_size(input_rank - 2); in Compile() local
491 const int64 num_rows = input_shape.dim_size(input_rank - 2); in Compile() local
/external/tensorflow/tensorflow/core/kernels/
Dbincount_op_gpu.cu.cc135 const int num_rows, const int num_cols, in BincountColReduceKernel()
156 const int num_rows, in BincountColReduceSharedKernel()
202 const int num_rows = in.dimension(0); in Compute() local
Dbincount_op.cc140 const int num_rows = out.dimension(0); in Compute() local
259 const int64 num_rows = data.dim_size(0); in Compute() local
358 const int64 num_rows = shape(0); in Compute() local
419 int num_rows = splits.size() - 1; in Compute() local
/external/webp/src/dec/
Dalpha_dec.c110 static int ALPHDecode(VP8Decoder* const dec, int row, int num_rows) { in ALPHDecode()
179 int row, int num_rows) { in VP8DecompressAlphaRows()
/external/tensorflow/tensorflow/python/ops/
Dlinalg_ops_impl.py33 def eye(num_rows, argument
/external/eigen/unsupported/test/
Dcxx11_tensor_complex_cuda.cu79 const int num_rows = internal::random<int>(1024, 5*1024); in test_cuda_sum_reductions() local
116 const int num_rows = internal::random<int>(1024, 5*1024); in test_cuda_product_reductions() local
/external/tensorflow/tensorflow/core/ops/
Dcheckpoint_ops.cc63 int64 num_rows; in __anon62d746c20202() local
/external/webp/src/dsp/
Dfilters.c47 int row, int num_rows, in DoHorizontalFilter_C()
84 int row, int num_rows, in DoVerticalFilter_C()
129 int row, int num_rows, in DoGradientFilter_C()
Dfilters_mips_dsp_r2.c198 int row, int num_rows, in DoHorizontalFilter_MIPSdspR2()
246 int row, int num_rows, in DoVerticalFilter_MIPSdspR2()
315 int row, int num_rows, uint8_t* out) { in DoGradientFilter_MIPSdspR2()
/external/tensorflow/tensorflow/core/kernels/sparse/
Dsparse_cholesky_op.cc83 int64 num_rows; in Compute() local
232 int64* num_rows) { in ValidateInputs()
Dkernels.cc30 const int64 batch_size, const int num_rows, in operator ()()
/external/tensorflow/tensorflow/compiler/xla/client/lib/
Dlogdet.cc59 auto num_rows = in LogDet() local
/external/libhevc/decoder/
Dihevcd_fmt_conv.c393 WORD32 num_rows, num_cols, src_strd, dst_strd; in ihevcd_fmt_conv_420sp_to_420sp() local
492 WORD32 num_rows, num_cols, src_strd, dst_strd; in ihevcd_fmt_conv_420sp_to_420sp_swap_uv() local
605 WORD32 num_rows, num_cols, src_strd, dst_strd; in ihevcd_fmt_conv_420sp_to_420p() local
698 WORD32 num_rows) in ihevcd_fmt_conv()
/external/tensorflow/tensorflow/python/kernel_tests/
Dbincount_op_test.py213 def _test_bincount_col_count(self, num_rows, num_cols, size, dtype): argument
226 def _test_bincount_col_binary(self, num_rows, num_cols, size, dtype): argument
242 def _test_bincount_col_count_with_weights(self, num_rows, num_cols, size, argument
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dconvert_matrix_diag_v2_or_v3_to_v1.cc60 int num_rows = input_num_rows.GetBuffer<ArrayDataType::kInt32>().data[0]; in Run() local
/external/libmpeg2/common/
Dideint_utils.c96 WORD32 num_rows) in ideint_weave_pic()
260 WORD32 num_cols, num_rows; in ideint_pad_blk() local

123456