Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h699 PoolingWorkspaceCache m_pooling_cache; variable
Drocm_dnn.cc4129 m_pooling_cache.find(input_data.opaque(), input_dimensions, in DoPoolForward()
4141 m_pooling_cache.insert(input_data.opaque(), input_dimensions, in DoPoolForward()
4233 m_pooling_cache.find(input_data.opaque(), input_dimensions, in DoPoolBackwardImpl()