Searched refs:m_pooling_cache (Results 1 – 2 of 2) sorted by relevance
636 PoolingWorkspaceCache m_pooling_cache; variable
4059 m_pooling_cache.find(input_data.opaque(), input_dimensions, in DoPoolForward()4070 m_pooling_cache.insert(input_data.opaque(), input_dimensions, in DoPoolForward()4215 m_pooling_cache.find(input_data.opaque(), input_dimensions, in DoPoolBackward()