Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1546 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1557 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1568 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
1580 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
1593 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
1606 const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
Dstream.cc617 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
639 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
659 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
680 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
700 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
727 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
752 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.cc3893 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3905 const dnn::PoolingDescriptor& pooling_dimensions, int _type) { in IsSame()
3920 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in find()
3938 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in insert()
3987 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4053 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4084 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackwardImpl()
4195 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4208 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4223 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.cc5470 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
5495 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
5520 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
5545 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
5570 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
5600 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
5630 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()