Searched refs:OutputMapper (Results 1 – 6 of 6) sorted by relevance
119 template <typename Scalar, typename IndexType, typename OutputMapper,124 template <typename IndexType, typename OutputMapper, bool ConjugateLhs,126 struct mkldnn_gemm_kernel</*Scalar*/ float, IndexType, OutputMapper,132 void operator()(const OutputMapper& output, const float* blockA,238 template <typename StorageIndex, typename OutputMapper, typename LhsMapper,240 struct TensorContractionKernel<float, float, float, StorageIndex, OutputMapper,252 using GemmKernel = mkldnn_gemm_kernel<Scalar, StorageIndex, OutputMapper>;264 gebp_kernel<Scalar, Scalar, StorageIndex, OutputMapper, Traits::mr,290 const OutputMapper& output_mapper, const Scalar* lhsBlock,
107 using OutputMapper = blas_data_mapper<Scalar, Index, ColMajor>; in TEST() typedef109 mkldnn_gemm_kernel<Scalar, Index, OutputMapper, ColMajor>; in TEST()113 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);