Home
last modified time | relevance | path

Searched refs:pooling_dimensions (Results 1 – 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_dnn.h427 const dnn::PoolingDescriptor& pooling_dimensions,
435 const dnn::PoolingDescriptor& pooling_dimensions,
443 const dnn::PoolingDescriptor& pooling_dimensions,
451 const dnn::PoolingDescriptor& pooling_dimensions,
459 const dnn::PoolingDescriptor& pooling_dimensions,
469 const dnn::PoolingDescriptor& pooling_dimensions,
479 const dnn::PoolingDescriptor& pooling_dimensions,
Dcuda_dnn.cc3729 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3741 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward()
3754 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3766 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward()
3779 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3792 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward()
3804 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3816 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolForward()
3829 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward() argument
3844 CudnnPoolingDescriptor pooling_desc(pooling_dimensions); in DoPoolBackward()
[all …]
/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_dnn.h418 const dnn::PoolingDescriptor& pooling_dimensions,
426 const dnn::PoolingDescriptor& pooling_dimensions,
434 const dnn::PoolingDescriptor& pooling_dimensions,
442 const dnn::PoolingDescriptor& pooling_dimensions,
452 const dnn::PoolingDescriptor& pooling_dimensions,
462 const dnn::PoolingDescriptor& pooling_dimensions,
Drocm_dnn.cc3476 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3486 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3500 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolForward()
3515 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward() argument
3530 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolForward()
3545 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward() argument
3558 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward() argument
3575 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolBackward()
3654 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward() argument
3671 ScopedPoolingDescriptor pooling_desc{pooling_dimensions}; in DoPoolBackward()
/external/tensorflow/tensorflow/stream_executor/
Ddnn.h1571 const dnn::PoolingDescriptor& pooling_dimensions,
1579 const dnn::PoolingDescriptor& pooling_dimensions,
1590 const dnn::PoolingDescriptor& pooling_dimensions,
1601 const dnn::PoolingDescriptor& pooling_dimensions,
1613 const dnn::PoolingDescriptor& pooling_dimensions,
1626 const dnn::PoolingDescriptor& pooling_dimensions,
1639 const dnn::PoolingDescriptor& pooling_dimensions,
Dstream.cc1221 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward() argument
1226 VLOG_CALL(PARAM(pooling_dimensions), PARAM(input_dimensions), in ThenPoolForward()
1232 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1246 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward() argument
1251 VLOG_CALL(PARAM(pooling_dimensions), PARAM(input_dimensions), in ThenPoolForward()
1257 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1268 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward() argument
1274 VLOG_CALL(PARAM(pooling_dimensions), PARAM(input_dimensions), in ThenPoolForward()
1280 CheckError(dnn->DoPoolForward(this, pooling_dimensions, input_dimensions, in ThenPoolForward()
1291 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward() argument
[all …]
Dstream.h517 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
524 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
531 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
538 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
545 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,
554 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,
563 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,