| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 502 PoolingDescriptor::PoolingDescriptor(int ndims) in PoolingDescriptor() function in stream_executor::dnn::PoolingDescriptor
510 PoolingDescriptor::PoolingDescriptor() : PoolingDescriptor(/*ndims=*/2) {} in PoolingDescriptor() function in stream_executor::dnn::PoolingDescriptor
512 void PoolingDescriptor::CloneFrom(const PoolingDescriptor& other) { in CloneFrom()
521 string PoolingDescriptor::ToString() const { in ToString()
540 string PoolingDescriptor::ToShortString() const { in ToShortString()
|
| D | dnn.h | 655 class PoolingDescriptor {
657 PoolingDescriptor();
658 explicit PoolingDescriptor(int ndims);
660 PoolingDescriptor& set_pooling_mode(PoolingMode value) {
664 PoolingDescriptor& set_window_height(int64 value) {
668 PoolingDescriptor& set_window_width(int64 value) {
672 PoolingDescriptor& set_window(DimIndex dim, int64 value) {
676 PoolingDescriptor& set_vertical_padding(int64 value) {
680 PoolingDescriptor& set_horizontal_padding(int64 value) {
684 PoolingDescriptor& set_padding(DimIndex dim, int64 value) {
[all …]
|
| D | stream.h | 517 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,
|
| D | stream.cc | 50 string ToVlogString(const dnn::PoolingDescriptor &descriptor) { in ToVlogString()
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()
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 427 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,
|
| D | cuda_dnn.cc | 423 using PoolingDescriptor = typedef
457 PoolingDescriptor CreatePoolingDescriptor() { in CreatePoolingDescriptor()
460 return PoolingDescriptor(result); in CreatePoolingDescriptor()
719 const dnn::PoolingDescriptor& pooling_descriptor) in CudnnPoolingDescriptor()
751 PoolingDescriptor handle_; // Owned.
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()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 418 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,
|
| D | rocm_dnn.cc | 64 using dnn::PoolingDescriptor;
708 ScopedPoolingDescriptor(const PoolingDescriptor& pooling_descriptor) in ScopedPoolingDescriptor()
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/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 71 se::dnn::PoolingDescriptor pooling_desc(3); in Compute()
189 se::dnn::PoolingDescriptor pooling_desc(3); in Compute()
|
| D | pooling_ops_common.cc | 228 se::dnn::PoolingDescriptor pooling_desc; in Compute()
376 se::dnn::PoolingDescriptor pooling_desc; in Compute()
|