Searched defs:pooling_dimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 3729 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3754 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3779 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3804 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3829 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3859 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3889 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3476 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3486 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3515 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3545 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3558 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3654 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1221 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1246 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1268 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1291 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1313 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1343 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1370 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
|