Home
last modified time | relevance | path

Searched refs:MklPrimitiveFactory (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_input_ops.cc239 class MklConvBwdInputPrimitiveFactory : public MklPrimitiveFactory<T> {
413 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled() && in Compute()
414 (MklPrimitiveFactory<T>::IsLegacyPlatform() || in Compute()
Dmkl_conv_grad_filter_ops.cc299 class MklConvBwdFilterPrimitiveFactory : public MklPrimitiveFactory<T> {
475 bool do_not_cache = MklPrimitiveFactory<T>::IsPrimitiveMemOptEnabled(); in Compute()
Dmkl_pooling_ops_common.h153 class MklPoolingFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
305 class MklPoolingBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
Dmkl_slice_op.cc242 class MklSlicePrimitiveFactory : public MklPrimitiveFactory<T> {
Dmkl_fused_batch_norm_op.cc223 class MklFusedBatchNormFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
447 class MklFusedBatchNormBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
Dmkl_relu_op.cc167 class MklEltwiseFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
381 class MklEltwiseBwdPrimitiveFactory : public MklPrimitiveFactory<T> {
Dmkl_conv_ops.cc321 class MklConvFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
1060 MklPrimitiveFactory<Tinput>::IsPrimitiveMemOptEnabled() && in Compute()
1062 (MklPrimitiveFactory<Tinput>::IsLegacyPlatform() || in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h2169 class MklPrimitiveFactory {
2171 MklPrimitiveFactory() {} in MklPrimitiveFactory() function
2173 ~MklPrimitiveFactory() {} in ~MklPrimitiveFactory()
2176 auto& lru_cache = MklPrimitiveFactory<T>::GetLRUCache(); in GetOp()
2181 auto& lru_cache = MklPrimitiveFactory<T>::GetLRUCache(); in SetOp()
2294 class MklReorderPrimitiveFactory : public MklPrimitiveFactory<T> {