Searched defs:pooling_dimensions (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 4271 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4296 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4321 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4346 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4371 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4401 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
4431 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 3765 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3775 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3804 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
3834 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3847 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
3943 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1363 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1388 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1410 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1433 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
1455 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1485 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
1512 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
|