| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 532 PoolingDescriptor::PoolingDescriptor(int ndims) in PoolingDescriptor() function in stream_executor::dnn::PoolingDescriptor
540 PoolingDescriptor::PoolingDescriptor() : PoolingDescriptor(/*ndims=*/2) {} in PoolingDescriptor() function in stream_executor::dnn::PoolingDescriptor
542 void PoolingDescriptor::CloneFrom(const PoolingDescriptor& other) { in CloneFrom()
551 std::string PoolingDescriptor::ToString() const { in ToString()
569 std::string PoolingDescriptor::ToShortString() const { in ToShortString()
|
| D | dnn.h | 653 class PoolingDescriptor {
655 PoolingDescriptor();
656 explicit PoolingDescriptor(int ndims);
658 PoolingDescriptor& set_pooling_mode(PoolingMode value) { in set_pooling_mode()
662 PoolingDescriptor& set_window_height(int64 value) { in set_window_height()
666 PoolingDescriptor& set_window_width(int64 value) { in set_window_width()
670 PoolingDescriptor& set_window(DimIndex dim, int64 value) { in set_window()
674 PoolingDescriptor& set_vertical_padding(int64 value) { in set_vertical_padding()
678 PoolingDescriptor& set_horizontal_padding(int64 value) { in set_horizontal_padding()
682 PoolingDescriptor& set_padding(DimIndex dim, int64 value) { in set_padding()
[all …]
|
| D | stream.h | 540 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
547 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
554 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
561 Stream &ThenPoolForward(const dnn::PoolingDescriptor &pooling_dimensions,
568 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,
577 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,
586 Stream &ThenPoolBackward(const dnn::PoolingDescriptor &pooling_dimensions,
|
| D | stream.cc | 49 std::string ToVlogString(const dnn::PoolingDescriptor &descriptor) { in ToVlogString()
820 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
842 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
862 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
883 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolForward()
903 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
930 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
955 const dnn::PoolingDescriptor &pooling_dimensions, in ThenPoolBackward()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.h | 45 dnn::PoolingDescriptor op;
52 const dnn::PoolingDescriptor& pooling_dimensions, int _type);
63 const dnn::PoolingDescriptor& pooling_dimensions, int _type,
67 const dnn::PoolingDescriptor& pooling_dimensions, int _type,
470 const dnn::PoolingDescriptor& pooling_dimensions,
478 const dnn::PoolingDescriptor& pooling_dimensions,
486 const dnn::PoolingDescriptor& pooling_dimensions,
494 const dnn::PoolingDescriptor& pooling_dimensions,
504 const dnn::PoolingDescriptor& pooling_dimensions,
514 const dnn::PoolingDescriptor& pooling_dimensions,
[all …]
|
| D | rocm_dnn.cc | 68 using dnn::PoolingDescriptor;
823 ScopedPoolingDescriptor(const PoolingDescriptor& pooling_descriptor) in ScopedPoolingDescriptor()
4002 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4014 const dnn::PoolingDescriptor& pooling_dimensions, int _type) { in IsSame()
4029 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in find()
4047 const dnn::PoolingDescriptor& pooling_dimensions, int _type, in insert()
4096 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4162 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4193 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackwardImpl()
4304 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.h | 446 const dnn::PoolingDescriptor& pooling_dimensions,
454 const dnn::PoolingDescriptor& pooling_dimensions,
462 const dnn::PoolingDescriptor& pooling_dimensions,
470 const dnn::PoolingDescriptor& pooling_dimensions,
478 const dnn::PoolingDescriptor& pooling_dimensions,
488 const dnn::PoolingDescriptor& pooling_dimensions,
498 const dnn::PoolingDescriptor& pooling_dimensions,
|
| D | cuda_dnn.cc | 436 using PoolingDescriptor = typedef
472 PoolingDescriptor CreatePoolingDescriptor() { in CreatePoolingDescriptor()
475 return PoolingDescriptor(result); in CreatePoolingDescriptor()
780 const dnn::PoolingDescriptor& pooling_descriptor) in CudnnPoolingDescriptor()
812 PoolingDescriptor handle_; // Owned.
4262 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4287 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4312 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4337 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolForward()
4362 Stream* stream, const dnn::PoolingDescriptor& pooling_dimensions, in DoPoolBackward()
[all …]
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.cc | 71 se::dnn::PoolingDescriptor pooling_desc(3); in Compute()
204 se::dnn::PoolingDescriptor pooling_desc(3); in Compute()
|
| D | pooling_ops_common.cc | 352 se::dnn::PoolingDescriptor pooling_desc; in Compute()
667 se::dnn::PoolingDescriptor pooling_desc; in Compute()
|