Searched refs:pack_lhs (Results 1 – 9 of 9) sorted by relevance
148 …gemm_pack_lhs<Scalar, Index, LhsMapper, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;193 …pack_lhs(blockA, LhsMapper(triangularBuffer.data(), triangularBuffer.outerStride()), actualPanelWi…204 … pack_lhs(blockA, lhs.getSubMapper(startTarget,startBlock), actualPanelWidth, lengthTarget);295 …gemm_pack_lhs<Scalar, Index, LhsMapper, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;357 pack_lhs(blockA, lhs.getSubMapper(i2, actual_k2), actual_kc, actual_mc);
79 …gemm_pack_lhs<Scalar, Index, TriMapper, Traits::mr, Traits::LhsProgress, TriStorageOrder> pack_lhs;160 … pack_lhs(blockA, tri.getSubMapper(startTarget,startBlock), actualPanelWidth, lengthTarget);177 pack_lhs(blockA, tri.getSubMapper(i2, IsLower ? k2 : k2-kc), actual_kc, actual_mc);
353 symm_pack_lhs<Scalar, Index, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;382 pack_lhs(blockA, &lhs(k2,k2), lhsStride, actual_kc, actual_mc);440 …gemm_pack_lhs<Scalar, Index, LhsMapper, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;453 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc);
78 …m_pack_lhs<LhsScalar, Index, LhsMapper, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;113 …pack_lhs(blockA+info[tid].lhs_start*actual_kc, lhs.getSubMapper(info[tid].lhs_start,k), actual_kc,…179 pack_lhs(blockA, lhs.getSubMapper(i2,k2), actual_kc, actual_mc);
87 …m_pack_lhs<LhsScalar, Index, LhsMapper, Traits::mr, Traits::LhsProgress, LhsStorageOrder> pack_lhs;103 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc);
473 void pack_lhs(Index m, Index k) {605 pack_lhs(start, k);969 LhsPacker pack_lhs;970 pack_lhs(arg.blockA, arg.lhs.getSubMapper(arg.m_start, arg.k_start), arg.kc, arg.mc);
466 …k_lhs<LhsScalar, Index, typename LhsMapper::SubMapper, mr, Traits::LhsProgress, ColMajor> pack_lhs;497 pack_lhs(blockA, lhs.getSubMapper(i2, k2), actual_kc, actual_mc, 0, 0);
505 pack_lhs; member in tensorflow::GemmFilterPacker514 void Run() { pack_lhs(lhs_block_, lhs_mapper_, depth_, rows_); } in Run()
1697 PackLhsImpl pack_lhs; in PackLhsHelper() local1733 pack_lhs(packed.data() + packed_offset, sub_mapper, rows, cols); in PackLhsHelper()1735 pack_lhs(packed.data() + packed_offset, sub_mapper, cols, rows); in PackLhsHelper()