Searched defs:pool_mode (Results 1 – 4 of 4) sorted by relevance
22 void PoolingGrad::Init(const PoolMode &pool_mode, const std::vector<int64_t> &window, in Init()36 void PoolingGrad::set_pool_mode(const PoolMode &pool_mode) { in set_pool_mode()
51 auto pool_mode = pooling_prim->poolingMode(); in PopulatePoolingParameter() local
50 std::ostream &operator<<(std::ostream &code, PoolMode pool_mode) { in operator <<()
25 OpParameter *CreateParameter(PoolMode pool_mode, int window_h, int window_w, int stride_h, int stri… in CreateParameter()