Searched refs:kAverage (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| 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_3d.cc | 747 DnnPooling3dOp<T>::Compute(context, se::dnn::PoolingMode::kAverage, window, in launch()
793 context, se::dnn::PoolingMode::kAverage, window, stride, padding, out, in launch()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | pooling.cc | 47 kAverage, enumerator
96 if (pool_type == kAverage || pool_type == kMax) { in GenericPrepare()
462 pooling::GenericPrepare<pooling::kAverage>, in Register_AVERAGE_POOL_REF()
483 pooling::Init, pooling::Free, pooling::GenericPrepare<pooling::kAverage>, in Register_AVERAGE_POOL_GENERIC_OPT()
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 185 case PoolingMode::kAverage: in ShortPoolingModeString()
|
| D | dnn.h | 627 kAverage, enumerator
|