Searched defs:pooling_dimensions (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3955 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3990 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4025 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4059 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4097 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4135 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1267 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1291 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1312 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1333 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1360 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1384 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
|