Add assertion for cpu backend shape conversion (#4904)
author오형석/On-Device Lab(SR)/Staff Engineer/삼성전자 <hseok82.oh@samsung.com>
Mon, 1 Apr 2019 02:08:47 +0000 (11:08 +0900)
committerGitHub Enterprise <noreply-CODE@samsung.com>
Mon, 1 Apr 2019 02:08:47 +0000 (11:08 +0900)
CPU backend assume that neurun's internal shape is always same or less than 4.
It is true on current neurun implementation, but it may change.
So add assertion to check this.

Signed-off-by: Hyeongseok Oh <hseok82.oh@samsung.com>
runtimes/neurun/backend/cpu/kernel/OperationUtils.cc

index c01770a..db59fa8 100644 (file)
@@ -195,6 +195,9 @@ Shape getShape(const ::neurun::model::operand::Object &o)
   shape.scale = o.typeInfo().scale();
   shape.offset = o.typeInfo().offset();
 
+  // CPU backend assume that neurun internal shape's rank is always same or less than 4
+  assert(shape.dimensions.size() <= 4);
+
   return shape;
 }