diff options
Diffstat (limited to 'src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp')
| -rw-r--r-- | src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp b/src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp index 431ef31437..4c67ec2c8e 100644 --- a/src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp +++ b/src/backends/backendsCommon/test/GatherEndToEndTestImpl.hpp @@ -42,6 +42,8 @@ void GatherEndToEnd(const std::vector<BackendId>& backends) paramsInfo.SetQuantizationScale(1.0f); paramsInfo.SetQuantizationOffset(0); + paramsInfo.SetConstant(true); + indicesInfo.SetConstant(true); outputInfo.SetQuantizationScale(1.0f); outputInfo.SetQuantizationOffset(0); @@ -78,6 +80,8 @@ void GatherMultiDimEndToEnd(const std::vector<BackendId>& backends) paramsInfo.SetQuantizationScale(1.0f); paramsInfo.SetQuantizationOffset(0); + paramsInfo.SetConstant(true); + indicesInfo.SetConstant(true); outputInfo.SetQuantizationScale(1.0f); outputInfo.SetQuantizationOffset(0); |