Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dpooling_test_util.cc38 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()
Dpooling.cc299 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()
/external/tensorflow/tensorflow/lite/delegates/gpu/gl/kernels/
Dpooling_test.cc58 attr.type = PoolingType::MAX; in TEST()
88 attr.type = PoolingType::MAX; in TEST()
114 attr.type = PoolingType::AVERAGE; in TEST()
Dpooling.cc180 case PoolingType::AVERAGE: in GenerateCode()
182 case PoolingType::MAX: in GenerateCode()
/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.cc125 template <typename Device, typename T, PoolingType Type>
/external/tensorflow/tensorflow/lite/delegates/gpu/common/
Doperations.h145 enum class PoolingType { enum
156 PoolingType type = PoolingType::UNDEFINED;
168 PoolingType type = PoolingType::UNDEFINED;
Dmodel_builder.cc1347 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()
/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
Dglobal_pooling_to_reduce_op_test.cc45 attr.type = tflite::gpu::PoolingType::AVERAGE; in TEST()
Dglobal_pooling_to_reduce_op.cc44 return attr.type == tflite::gpu::PoolingType::AVERAGE && in IsGlobalAveragePooling()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_nodes.cc3423 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()