HomeSort by relevance Sort by last modified time
    Searched refs:MklPoolingFwdPrimitiveFactory (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/core/kernels/
mkl_pooling_ops_common.h 153 class MklPoolingFwdPrimitiveFactory : public MklPrimitiveFactory<T> {
160 MklPoolingFwdPrimitiveFactory<T>::GetInstance().GetPoolingFwd(
165 MklPoolingFwdPrimitiveFactory<T>::GetInstance().SetPoolingFwd(
171 static MklPoolingFwdPrimitiveFactory& GetInstance() {
172 static MklPoolingFwdPrimitiveFactory instance_;
177 MklPoolingFwdPrimitiveFactory() {}
178 ~MklPoolingFwdPrimitiveFactory() {}
mkl_avgpooling_op.cc 508 pooling_fwd = MklPoolingFwdPrimitiveFactory<T>::Get(fwdParams);
mkl_maxpooling_op.cc 584 pooling_fwd = MklPoolingFwdPrimitiveFactory<T>::Get(fwdParams);
    [all...]

Completed in 544 milliseconds