Home
last modified time | relevance | path

Searched refs:PoolingType (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dpooling_ops_3d.h25 enum PoolingType { MAX, AVG }; enum
27 template <typename Device, typename T, PoolingType Type>
Dpooling_ops_3d.cc130 template <typename Device, typename T, PoolingType Type>
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc2645 nvinfer1::PoolingType type; in ConvertPool()
2647 type = nvinfer1::PoolingType::kMAX; in ConvertPool()
2649 type = nvinfer1::PoolingType::kAVERAGE; in ConvertPool()