diff options
Diffstat (limited to 'tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h')
-rw-r--r-- | tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h | 2 |
1 files changed, 2 insertions, 0 deletions
diff --git a/tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h b/tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h index edf0dff54b..08fffb305b 100644 --- a/tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h +++ b/tests/validation/fixtures/dynamic_fusion/operators/CastFixture.h @@ -120,6 +120,8 @@ protected: GpuWorkloadSketch sketch{&context}; // Create sketch tensors + // Here, we use DataLayout::NCHW just for the test. However, the optimal data layout to + // be used with dynamic fusion is NHWC ITensorInfo *src_info = context.create_tensor_info(TensorInfo(shape, 1, dt_in, DataLayout::NCHW)); // layout is not important ITensorInfo *dst_info = context.create_tensor_info(); |