aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h')
-rw-r--r--tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h41
1 files changed, 8 insertions, 33 deletions
diff --git a/tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h b/tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h
index dd3519b549..b0c7143d91 100644
--- a/tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h
+++ b/tests/validation/fixtures/dynamic_fusion/gpu/cl/Pool2dFixture.h
@@ -51,11 +51,11 @@ template <typename TensorType, typename AccessorType, typename FunctionType, typ
class DynamicFusionGpuPool2dValidationGenericFixture : public framework::Fixture
{
public:
- void setup(TensorShape input_shape, const Pool2dAttributes &pool_attr, DataType data_type, bool mixed_precision)
+ void setup(TensorShape input_shape, const Pool2dAttributes &pool_attr, DataType data_type)
{
- _target = compute_target(input_shape, pool_attr, data_type, mixed_precision);
- _reference =
- compute_reference(input_shape, convert_pool_attr_to_pool_info(pool_attr, mixed_precision), data_type);
+ _target = compute_target(input_shape, pool_attr, data_type);
+ _reference = compute_reference(
+ input_shape, convert_pool_attr_to_pool_info(pool_attr, true /* mixed_precision */), data_type);
}
protected:
@@ -82,10 +82,7 @@ protected:
}
// Given input is in nchw format
- TensorType compute_target(TensorShape input_shape,
- const Pool2dAttributes &pool_attr,
- const DataType data_type,
- bool mixed_precision)
+ TensorType compute_target(TensorShape input_shape, const Pool2dAttributes &pool_attr, const DataType data_type)
{
CLScheduler::get().default_reinit();
@@ -102,7 +99,7 @@ protected:
auto dst_info = context.create_tensor_info();
// Create Pool2dSettings
- GpuPool2dSettings pool_settings = GpuPool2dSettings().mixed_precision(mixed_precision);
+ GpuPool2dSettings pool_settings = GpuPool2dSettings();
ITensorInfo *ans_info = FunctionType::create_op(sketch, input_info, pool_attr, pool_settings);
GpuOutput::create_op(sketch, ans_info, dst_info);
@@ -168,29 +165,7 @@ public:
input_shape,
Pool2dAttributes().pool_type(pool_type).pool_size(pool_size).pad(pad).stride(stride).exclude_padding(
exclude_padding),
- data_type, false);
- }
-};
-
-template <typename TensorType, typename AccessorType, typename FunctionType, typename T>
-class DynamicFusionGpuPool2dMixedPrecisionValidationFixture
- : public DynamicFusionGpuPool2dValidationGenericFixture<TensorType, AccessorType, FunctionType, T>
-{
-public:
- void setup(TensorShape input_shape,
- PoolingType pool_type,
- Size2D pool_size,
- Padding2D pad,
- Size2D stride,
- bool exclude_padding,
- DataType data_type,
- bool mixed_precision)
- {
- DynamicFusionGpuPool2dValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(
- input_shape,
- Pool2dAttributes().pool_type(pool_type).pool_size(pool_size).pad(pad).stride(stride).exclude_padding(
- exclude_padding),
- data_type, mixed_precision);
+ data_type);
}
};
@@ -202,7 +177,7 @@ public:
void setup(TensorShape input_shape, Pool2dAttributes pool_attr, DataType data_type)
{
DynamicFusionGpuPool2dValidationGenericFixture<TensorType, AccessorType, FunctionType, T>::setup(
- input_shape, pool_attr, data_type, false);
+ input_shape, pool_attr, data_type);
}
};