Home
last modified time | relevance | path

Searched defs:is_pool2d (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_avgpooling_op.cc63 bool is_pool2d = (this->ksize_.size() == 4); in Compute() local
210 bool is_pool2d = (this->ksize_.size() == 4); in Compute() local
H A Dmkl_maxpooling_op.cc65 bool is_pool2d = (this->ksize_.size() == 4); in Compute() local
262 bool is_pool2d = (this->ksize_.size() == 4); in Compute() local
H A Dmkl_pooling_ops_common.cc288 bool is_pool2d = (ksize.size() == 4); in Init() local
H A Dmkl_pooling_ops_common.h472 bool is_pool2d = (this->ksize_.size() == 4); in MklPoolingOpBase() local
529 bool is_pool2d) { in PoolParamsToDims()