Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1645 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1656 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1667 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1679 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
1692 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
1705 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
Dstream.cc820 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
842 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
862 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
883 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
903 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
930 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
955 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc4002 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4014 const dnn::PoolingDescriptor& pooling_dimensions, int _type) { in IsSame()
4029 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in find()
4047 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in insert()
4096 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4162 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4193 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackwardImpl()
4304 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4317 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4332 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc4262 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4287 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4312 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4337 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4362 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4392 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4422 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()