diff options
Diffstat (limited to 'src/core/TensorInfo.cpp')
| -rw-r--r-- | src/core/TensorInfo.cpp | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/src/core/TensorInfo.cpp b/src/core/TensorInfo.cpp index 24988e2217..bd0c85f3d4 100644 --- a/src/core/TensorInfo.cpp +++ b/src/core/TensorInfo.cpp @@ -348,7 +348,7 @@ ITensorInfo &TensorInfo::set_format(Format format) return *this; } -ITensorInfo &TensorInfo::set_tensor_shape(TensorShape shape) +ITensorInfo &TensorInfo::set_tensor_shape(const TensorShape &shape) { _tensor_shape = shape; _offset_first_element_in_bytes = 0; @@ -378,7 +378,7 @@ ITensorInfo &TensorInfo::set_fixed_point_position(int fixed_point_position) return *this; } -ITensorInfo &TensorInfo::set_quantization_info(QuantizationInfo quantization_info) +ITensorInfo &TensorInfo::set_quantization_info(const QuantizationInfo &quantization_info) { _quantization_info = quantization_info; return *this; |