Searched refs:PoolingType (Results 1 – 11 of 11) sorted by relevance
38 attr.type = PoolingType::AVERAGE; in AveragePoolingTest()69 attr.type = PoolingType::AVERAGE; in AveragePoolingNonEmptyPaddingTest()101 attr.type = PoolingType::MAX; in MaxPoolingTest()132 attr.type = PoolingType::MAX; in MaxPoolingIndicesTest()
299 if (attr.type == PoolingType::AVERAGE) { in CreatePooling()301 } else if (attr.type == PoolingType::MAX) { in CreatePooling()323 if (attr.type == PoolingType::AVERAGE) { in CreatePooling()325 } else if (attr.type == PoolingType::MAX) { in CreatePooling()
58 attr.type = PoolingType::MAX; in TEST()88 attr.type = PoolingType::MAX; in TEST()114 attr.type = PoolingType::AVERAGE; in TEST()
180 case PoolingType::AVERAGE: in GenerateCode()182 case PoolingType::MAX: in GenerateCode()
25 enum PoolingType { MAX, AVG }; enum27 template <typename Device, typename T, PoolingType Type>
125 template <typename Device, typename T, PoolingType Type>
145 enum class PoolingType { enum156 PoolingType type = PoolingType::UNDEFINED;168 PoolingType type = PoolingType::UNDEFINED;
1347 explicit Pooling2DOperationParser(PoolingType type) : type_(type) {} in Pooling2DOperationParser()1374 const PoolingType type_;2326 return std::make_unique<Pooling2DOperationParser>(PoolingType::AVERAGE); in NewOperationParser()2363 return std::make_unique<Pooling2DOperationParser>(PoolingType::MAX); in NewOperationParser()
45 attr.type = tflite::gpu::PoolingType::AVERAGE; in TEST()
44 return attr.type == tflite::gpu::PoolingType::AVERAGE && in IsGlobalAveragePooling()
3423 nvinfer1::PoolingType type; in ConvertPool3D()3425 type = nvinfer1::PoolingType::kMAX; in ConvertPool3D()3427 type = nvinfer1::PoolingType::kAVERAGE; in ConvertPool3D()3683 nvinfer1::PoolingType type; in ConvertPool()3685 type = nvinfer1::PoolingType::kMAX; in ConvertPool()3687 type = nvinfer1::PoolingType::kAVERAGE; in ConvertPool()