aboutsummaryrefslogtreecommitdiff
path: root/tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h
diff options
context:
space:
mode:
Diffstat (limited to 'tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h')
-rw-r--r--tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h10
1 files changed, 5 insertions, 5 deletions
diff --git a/tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h b/tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h
index a427d814cb..0d3b1f0296 100644
--- a/tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h
+++ b/tests/validation/fixtures/dynamic_fusion/operators/ReshapeFixture.h
@@ -75,13 +75,13 @@ protected:
GpuWorkloadSketch sketch{ &gpu_ctx };
// Create sketch tensors
- auto src_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, data_type));
- auto ans_info = sketch.create_tensor_info(TensorInfo(output_shape, 1, data_type));
- auto dst_info = sketch.create_tensor_info(TensorInfo(output_shape, 1, data_type));
+ TensorInfo src_info = sketch.create_tensor_info(TensorInfo(input_shape, 1, data_type));
+ TensorInfo dst_info = sketch.create_tensor_info(TensorInfo(output_shape, 1, data_type));
ReshapeAttributes attributes;
attributes.shape(output_shape);
- FunctionType::create_op(sketch, &src_info, &ans_info, attributes);
- GpuOutput::create_op(sketch, &ans_info, &dst_info);
+
+ ITensorInfo *ans_info = FunctionType::create_op(sketch, &src_info, attributes);
+ GpuOutput::create_op(sketch, ans_info, &dst_info);
// Configure runtime
ClWorkloadRuntime runtime;