| /external/tensorflow/tensorflow/lite/kernels/ |
| D | matrix_set_diag.cc | 65 const int row_size, const int col_size) { in FillDiagImpl() argument
69 for (int j = 0; j < col_size; ++j) { in FillDiagImpl()
72 out[i * col_size + j] = diag[idx]; in FillDiagImpl()
75 out[i * col_size + j] = in[i * col_size + j]; in FillDiagImpl()
79 out += row_size * col_size; in FillDiagImpl()
80 in += row_size * col_size; in FillDiagImpl()
87 const int col_size) { in FillDiag() argument
89 GetTensorData<T>(output), batch_size, row_size, col_size); in FillDiag()
103 const int col_size = output->dims->data[num_output_dims - 1]; in FillDiagHelper() local
107 col_size); in FillDiagHelper()
[all …]
|
| D | matrix_diag.cc | 65 const int col_size) { in FillDiagImpl() argument
69 for (int j = 0; j < col_size; ++j) { in FillDiagImpl()
72 out[i * col_size + j] = in[idx]; in FillDiagImpl()
75 out[i * col_size + j] = 0; 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()
99 const int col_size = output->dims->data[num_output_dims - 1]; 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()
[all …]
|
| D | embedding_lookup.cc | 103 int col_size = 1; in EvalHybrid() local
105 col_size *= SizeOfDimension(value, i); in EvalHybrid()
124 for (int j = 0; j < col_size; j++) { in EvalHybrid()
125 output_ptr[j + i * col_size] = in EvalHybrid()
126 value_ptr[j + idx * col_size] * scaling_factor; in EvalHybrid()
|
| /external/libhevc/encoder/ |
| D | ihevce_deblk.c | 288 WORD32 col_size; in ihevce_deblk_ctb() local
477 col_size = ctb_size / 4; in ihevce_deblk_ctb()
487 col_size += last_col; in ihevce_deblk_ctb()
491 col_size -= 1; in ihevce_deblk_ctb()
502 WORD32 col_size_temp = col_size; in ihevce_deblk_ctb()
577 col_size = ctb_size / 8; in ihevce_deblk_ctb()
588 col_size += last_col; in ihevce_deblk_ctb()
592 col_size--; in ihevce_deblk_ctb()
604 WORD32 col_size_temp = col_size; in ihevce_deblk_ctb()
|
| /external/oboe/samples/RhythmGame/third_party/glm/detail/ |
| D | type_gentype.inl | 11 typename base<vT, cT, rT, pT>::size_type base<vT, cT, rT, pT>::col_size() function in glm::detail::base
64 i < base<vT, cT, rT, pT>::col_size();
83 i < base<vT, cT, rT, pT>::col_size();
126 i < base<vT, cT, rT, pT>::col_size();
144 i < base<vT, cT, rT, pT>::col_size();
192 typename base<vT, cT, rT, pT>::size_type stop_col = x.col_size();
208 typename base<vT, cT, rT, pT>::size_type stop_col = x.col_size();
224 typename base<vT, cT, rT, pT>::size_type stop_col = x.col_size();
240 typename base<vT, cT, rT, pT>::size_type stop_col = x.col_size();
256 typename base<vT, cT, rT, pT>::size_type stop_col = x.col_size();
[all …]
|
| /external/libaom/libaom/av1/encoder/x86/ |
| D | av1_txfm1d_sse4.h | 82 const int col_size = txfm_size / num_per_128; in transpose_32() local
87 for (c = 0; c < col_size; c++) { in transpose_32()
88 transpose_32_4x4(col_size, &input[r * col_size + c], in transpose_32()
89 &output[c * 4 * col_size + r / 4]); in transpose_32()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | training_ops_gpu.cu.cc | 118 Tindex col_size = updates_size / indices_size; in SparseApplyAdagradKernel() local
120 Tindex indices_row = grad_index / col_size; in SparseApplyAdagradKernel()
128 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyAdagradKernel()
157 Tindex col_size = updates_size / indices_size; in SparseApplyProximalAdagradKernel() local
159 Tindex indices_row = grad_index / col_size; in SparseApplyProximalAdagradKernel()
167 Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyProximalAdagradKernel()
200 const Tindex col_size = updates_size / indices_size; in SparseApplyFtrlKernel() local
202 const Tindex indices_row = grad_index / col_size; in SparseApplyFtrlKernel()
210 const Tindex param_index = param_row * col_size + (grad_index % col_size); in SparseApplyFtrlKernel()
302 Tindex col_size = updates_size / indices_size; in SparseApplyKerasMomentumKernel() local
[all …]
|
| D | slice_op.cc | 179 const int64 col_size = size[1]; in Compute() local
191 col_size * sizeof(T)); in Compute()
|
| /external/jemalloc_new/src/ |
| D | stats.c | 380 col_size.size_val = reg_size; in stats_arena_bins_print()
479 col_size.size_val = lextent_size; in stats_arena_lextents_print()
|
| /external/mesa3d/src/compiler/ |
| D | glsl_types.cpp | 2588 unsigned col_size, col_align; in get_explicit_type_for_size_align() local
2589 type_info(this->column_type(), &col_size, &col_align); in get_explicit_type_for_size_align()
2590 unsigned stride = align(col_size, col_align); in get_explicit_type_for_size_align()
|
| /external/libaom/libaom/av1/encoder/ |
| D | bitstream.c | 3578 uint32_t col_size = total_size - col_offset - 4; in write_tiles_in_tg_obus() local
3579 mem_put_le32(dst + col_offset + tg_hdr_size, col_size); in write_tiles_in_tg_obus()
3582 max_tile_col_size = AOMMAX(max_tile_col_size, col_size); in write_tiles_in_tg_obus()
|