Searched refs:OutputMapper (Results 1 – 6 of 6) sorted by relevance
126 template <typename Scalar, typename IndexType, typename OutputMapper,131 template <typename IndexType, typename OutputMapper, bool ConjugateLhs,133 struct dnnl_gemm_kernel</*Scalar*/ float, IndexType, OutputMapper, ConjugateLhs,145 void operator()(const OutputMapper& output, const LhsScalar* blockA,193 template <typename IndexType, typename OutputMapper, bool ConjugateLhs = false,206 void operator()(const OutputMapper& output, const LhsScalar* blockA,507 typename StorageIndex, typename OutputMapper>513 template <typename StorageIndex, typename OutputMapper>514 struct GemmKernelProvider<float, float, float, StorageIndex, OutputMapper> {516 using GemmKernel = dnnl_gemm_kernel<float, StorageIndex, OutputMapper>;[all …]
114 using OutputMapper = blas_data_mapper<Scalar, Index, ColMajor>; in TEST() typedef116 dnnl_gemm_kernel<Scalar, Index, OutputMapper, ColMajor>; in TEST()120 OutputMapper output_mapper(mkldnn_result.data(), m); in TEST()
578 OutputMapper; typedef in tensorflow::GemmState584 Eigen::internal::gebp_kernel<T, T, int64, OutputMapper, Traits::mr,617 OutputMapper out_mapper_;
20 typename RhsMapper, typename OutputMapper, bool needs_edge_check>23 const OutputMapper output, Scalar* lhs_shmem, Scalar* rhs_shmem, in EigenContractionKernelInternal()500 typename RhsMapper, typename OutputMapper>504 const OutputMapper output, in EigenContractionKernel()516 …EigenContractionKernelInternal<Scalar, Index, LhsMapper, RhsMapper, OutputMapper, false>(lhs, rhs,… in EigenContractionKernel()518 …EigenContractionKernelInternal<Scalar, Index, LhsMapper, RhsMapper, OutputMapper, true>(lhs, rhs, … in EigenContractionKernel()524 typename RhsMapper, typename OutputMapper, bool CHECK_LHS_BOUNDARY,528 const OutputMapper output, float2 lhs_shmem2[][16], in EigenFloatContractionKernelInternal16x16()761 typename RhsMapper, typename OutputMapper, bool CHECK_LHS_BOUNDARY,765 const OutputMapper output, float2 lhs_shmem2[][32], in EigenFloatContractionKernelInternal()[all …]
31 template<typename LhsScalar, typename RhsScalar, typename RhsMapper, typename OutputMapper, typenam…36 OutputMapper& output;138 typedef internal::blas_data_mapper<Scalar, Index, ColMajor> OutputMapper;146 typedef internal::gebp_kernel<LhsScalar, RhsScalar, Index, OutputMapper,292 OutputMapper>(this->m_device, num_threads, lhs, rhs, buffer, m, n,300 typename LhsMapper, typename RhsMapper, typename OutputMapper>403 OutputMapper output_;772 typedef internal::blas_data_mapper<Scalar, Index, ColMajor> OutputMapper;780 typedef internal::gebp_kernel<LhsScalar, RhsScalar, Index, OutputMapper,784 …typedef internal::packRhsAndKernelArg<LhsScalar, RhsScalar, RhsMapper, OutputMapper, Index> packRK…[all …]
463 typedef internal::blas_data_mapper<Scalar, Index, ColMajor> OutputMapper;469 internal::gebp_kernel<LhsScalar, RhsScalar, Index, OutputMapper, mr, nr, false, false> gebp;478 OutputMapper output(buffer, m);