Searched refs:MklPrimitiveFactory (Results 1 – 12 of 12) sorted by relevance
241 class MklConvBwdInputPrimitiveFactory : public MklPrimitiveFactory<T> {429 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled() && in Compute()430 (MklPrimitiveFactory<T>::IsLegacyPlatform() || in Compute()
144 class MklSoftmaxPrimitiveFactory : public MklPrimitiveFactory<T> {
295 class MklConvBwdFilterPrimitiveFactory : public MklPrimitiveFactory<T> {473 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled(); in Compute()
154 class MklPoolingFwdPrimitiveFactory : public MklPrimitiveFactory<T> {299 class MklPoolingBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
292 class MklDnnMatMulFwdPrimitiveFactory : public MklPrimitiveFactory<T> {636 class MklMatMulPrimitiveFactory : public MklPrimitiveFactory<T> {
156 class MklReorderWithScalePrimitiveFactory : public MklPrimitiveFactory<T> {
250 class MklSlicePrimitiveFactory : public MklPrimitiveFactory<T> {
165 class MklEltwiseFwdPrimitiveFactory : public MklPrimitiveFactory<T> {376 class MklEltwiseBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
321 class MklConvFwdPrimitiveFactory : public MklPrimitiveFactory<float> {641 MklPrimitiveFactory<Tinput>::IsPrimitiveMemOptEnabled() && in Compute()643 (MklPrimitiveFactory<Tinput>::IsLegacyPlatform() || in Compute()
326 class MklFusedBatchNormFwdPrimitiveFactory : public MklPrimitiveFactory<T> {584 class MklFusedBatchNormBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
382 class MklConcatFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
1790 class MklPrimitiveFactory {1792 MklPrimitiveFactory() {} in MklPrimitiveFactory() function1794 ~MklPrimitiveFactory() {} in ~MklPrimitiveFactory()1797 auto& lru_cache = MklPrimitiveFactory<T>::GetLRUCache(); in GetOp()1802 auto& lru_cache = MklPrimitiveFactory<T>::GetLRUCache(); in SetOp()1902 class MklReorderPrimitiveFactory : public MklPrimitiveFactory<T> {