Home
last modified time | relevance | path

Searched refs:getSubMapper (Results 1 – 13 of 13) sorted by relevance

/external/eigen/Eigen/src/Core/products/
DTriangularMatrixMatrix.h165 pack_rhs(blockB, rhs.getSubMapper(actual_k2,0), actual_kc, cols);
195 gebp_kernel(res.getSubMapper(startBlock, 0), blockA, blockB,
204 … pack_lhs(blockA, lhs.getSubMapper(startTarget,startBlock), actualPanelWidth, lengthTarget);
206 gebp_kernel(res.getSubMapper(startTarget, 0), blockA, blockB,
220 (blockA, lhs.getSubMapper(i2, actual_k2), actual_kc, actual_mc);
222 gebp_kernel(res.getSubMapper(i2, 0), blockA, blockB, actual_mc,
321 pack_rhs(geb, rhs.getSubMapper(actual_k2,IsLower ? 0 : k2), actual_kc, rs);
334 rhs.getSubMapper(actual_k2+panelOffset, actual_j2),
357 pack_lhs(blockA, lhs.getSubMapper(i2, actual_k2), actual_kc, actual_mc);
368 gebp_kernel(res.getSubMapper(i2, actual_k2 + j2),
[all …]
DTriangularSolverMatrix.h153 …pack_rhs(blockB+actual_kc*j2, other.getSubMapper(startBlock,j2), actualPanelWidth, actual_cols, ac…
160 … pack_lhs(blockA, tri.getSubMapper(startTarget,startBlock), actualPanelWidth, lengthTarget);
162 …gebp_kernel(other.getSubMapper(startTarget,j2), blockA, blockB+actual_kc*j2, lengthTarget, actualP…
177 pack_lhs(blockA, tri.getSubMapper(i2, IsLower ? k2 : k2-kc), actual_kc, actual_mc);
179 …gebp_kernel(other.getSubMapper(i2, 0), blockA, blockB, actual_mc, actual_kc, cols, Scalar(-1), -1,…
245 if (rs>0) pack_rhs(geb, rhs.getSubMapper(actual_k2,startPanel), actual_kc, rs);
259 rhs.getSubMapper(actual_k2+panelOffset, actual_j2),
287 gebp_kernel(lhs.getSubMapper(i2,absolute_j2),
324 gebp_kernel(lhs.getSubMapper(i2, startPanel), blockA, geb,
DGeneralMatrixMatrix.h102 pack_rhs(blockB, rhs.getSubMapper(k,0), actual_kc, nc);
113 …pack_lhs(blockA+info[tid].lhs_start*actual_kc, lhs.getSubMapper(info[tid].lhs_start,k), actual_kc,…
131 …gebp(res.getSubMapper(info[i].lhs_start, 0), blockA+info[i].lhs_start*actual_kc, blockB, info[i].l…
140 pack_rhs(blockB, rhs.getSubMapper(k,j), actual_kc, actual_nc);
143 gebp(res.getSubMapper(0, j), blockA, blockB, rows, actual_kc, actual_nc, alpha);
179 pack_lhs(blockA, lhs.getSubMapper(i2,k2), actual_kc, actual_mc);
190 pack_rhs(blockB, rhs.getSubMapper(k2,j2), actual_kc, actual_nc);
193 gebp(res.getSubMapper(i2, j2), blockA, blockB, actual_mc, actual_kc, actual_nc, alpha);
DGeneralMatrixMatrixTriangular.h97 pack_rhs(blockB, rhs.getSubMapper(k2,0), actual_kc, size);
103 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc);
110 gebp(res.getSubMapper(i2, 0), blockA, blockB, actual_mc, actual_kc,
119 gebp(res.getSubMapper(i2, j2), blockA, blockB+actual_kc*j2, actual_mc,
161 gebp_kernel(res.getSubMapper(0, j), blockA, actual_b, j, depth, actualBlockSize, alpha,
184 gebp_kernel(res.getSubMapper(i, j), blockA+depth*i, actual_b, size-i,
DSelfadjointMatrixMatrix.h364 pack_rhs(blockB, rhs.getSubMapper(k2,0), actual_kc, cols);
374 pack_lhs_transposed(blockA, lhs_transpose.getSubMapper(i2, k2), actual_kc, actual_mc);
376 gebp_kernel(res.getSubMapper(i2, 0), blockA, blockB, actual_mc, actual_kc, cols, alpha);
384 gebp_kernel(res.getSubMapper(k2, 0), blockA, blockB, actual_mc, actual_kc, cols, alpha);
391 (blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc);
393 gebp_kernel(res.getSubMapper(i2, 0), blockA, blockB, actual_mc, actual_kc, cols, alpha);
453 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc);
455 gebp_kernel(res.getSubMapper(i2, 0), blockA, blockB, actual_mc, actual_kc, cols, alpha);
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorContractionThreadPool.h477 lhs_.getSubMapper(m1 * bm_, k * bk_), bk(k), bm(m1));
501 rhs_.getSubMapper(k * bk_, n1 * bn_), bk(k), bn(n1));
521 GebpKernel()(output_.getSubMapper(m1 * bm_, n1 * bn_),
529 GebpKernel()(output_.getSubMapper(m1 * bm_, n1 * bn_),
970 pack_lhs(arg.blockA, arg.lhs.getSubMapper(arg.m_start, arg.k_start), arg.kc, arg.mc);
986 pack_rhs(arg.blockB, arg.rhs.getSubMapper(arg.k, arg.n), arg.kc, arg.nc);
996 gebp(arg.output.getSubMapper(m_base_start, arg.n),
DTensorContraction.h497 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc, 0, 0);
503 pack_rhs(blockB, rhs.getSubMapper(k2, j2), actual_kc, actual_nc, 0, 0);
507 …gebp(output.getSubMapper(i2, j2), blockA, blockB, actual_mc, actual_kc, actual_nc, Scalar(1), -1, …
DTensorContractionMapper.h453 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const {
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc73 data_mapper.getSubMapper(slice_start[0], slice_start[1]), in TEST()
Deigen_spatial_convolutions_test.cc1542 input_mappers[input_idx].getSubMapper(depth_offset, col_offset); in PackRhsHelper()
1701 input_mappers[filter_idx].getSubMapper(row_offset, col_offset); in PackLhsHelper()
Deigen_spatial_convolutions-inl.h215 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper() function
Deigen_cuboid_convolution.h282 EIGEN_STRONG_INLINE SubMapper getSubMapper(Index i, Index j) const { in getSubMapper() function
/external/eigen/Eigen/src/Core/util/
DBlasUtil.h203 getSubMapper(Index i, Index j) const {
260 …EIGEN_ALWAYS_INLINE const_blas_data_mapper<Scalar, Index, StorageOrder> getSubMapper(Index i, Inde…