Searched refs:PoolingMode (Results 1 – 11 of 11) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | cudnn_pooling_gpu.h | 41 se::dnn::PoolingMode pooling_mode,
55 se::dnn::PoolingMode pooling_mode,
|
| D | pooling_ops_common_gpu.h | 43 se::dnn::PoolingMode pooling_mode,
59 se::dnn::PoolingMode pooling_mode,
|
| D | cudnn_pooling_gpu.cc | 35 se::dnn::PoolingMode pooling_mode, in Compute()
134 OpKernelContext* context, se::dnn::PoolingMode pooling_mode, in Compute()
140 CHECK((pooling_mode != se::dnn::PoolingMode::kMaximum) || in Compute()
|
| D | avgpooling_op.cc | 186 DnnPoolingOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, ksize_, in Compute()
192 DnnPoolingOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, ksize_, in Compute()
465 context, se::dnn::PoolingMode::kAverage, ksize_, stride_, padding_, in Compute()
550 DnnPoolingGradOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, in Compute()
607 DnnPoolingGradOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, in Compute()
|
| D | pooling_ops_common.cc | 214 se::dnn::PoolingMode pooling_mode, in Compute()
442 OpKernelContext* context, se::dnn::PoolingMode pooling_mode, in Compute()
448 CHECK((pooling_mode != se::dnn::PoolingMode::kMaximum) || in Compute()
|
| D | pooling_ops_3d.cc | 747 DnnPooling3dOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, window, in launch()
760 DnnPooling3dOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, window, in launch()
775 DnnPooling3dGradOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, in launch()
793 context, se::dnn::PoolingMode::kAverage, window, stride, padding, out, in launch()
|
| D | maxpooling_op.cc | 429 DnnPoolingGradOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, ksize, in Compute()
1211 DnnPoolingOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, ksize_, in Compute()
1218 DnnPoolingOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, ksize_, in Compute()
1332 DnnPoolingOp<T>::Compute(context, se::dnn::PoolingMode::kMaximum, ksize, in Compute()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 181 std::string ShortPoolingModeString(PoolingMode mode) { in ShortPoolingModeString()
183 case PoolingMode::kMaximum: in ShortPoolingModeString()
185 case PoolingMode::kAverage: in ShortPoolingModeString()
533 : mode_(dnn::PoolingMode::kMaximum), in PoolingDescriptor()
553 mode_ == dnn::PoolingMode::kMaximum ? "kMaximum" : "kAverage"; in ToString()
576 return absl::StrCat(mode_ == dnn::PoolingMode::kMaximum ? "max" : "avg", in ToShortString()
|
| D | dnn.h | 625 enum class PoolingMode : int64 { enum
638 std::string ShortPoolingModeString(PoolingMode mode);
658 PoolingDescriptor& set_pooling_mode(PoolingMode value) { in set_pooling_mode()
713 PoolingMode mode() const { return mode_; } in mode()
730 PoolingMode mode_;
|
| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_dnn.cc | 802 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in CudnnPoolingDescriptor()
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_dnn.cc | 848 (pooling_descriptor.mode() == dnn::PoolingMode::kMaximum in ScopedPoolingDescriptor()
|