Home
last modified time | relevance | path

Searched refs:m_pooling_cache_allowed (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc606 if (enable_pooling_cache) m_pooling_cache_allowed = true; in MIOpenSupport()
4128 m_pooling_cache_allowed && in DoPoolForward()
4200 if (m_pooling_cache_allowed) m_pooling_cache_enabled = true; in DoPoolBackwardImpl()
4232 bool cache_hit = m_pooling_cache_allowed && in DoPoolBackwardImpl()
Drocm_dnn.h700 bool m_pooling_cache_allowed = false; variable