diff options
| author | Tai Ly <tai.ly@arm.com> | 2024-02-14 22:35:44 +0000 |
|---|---|---|
| committer | Tai Ly <tai.ly@arm.com> | 2024-02-27 11:17:44 -0800 |
| commit | 8ead6c48d878346dfadc7fb48ee9ec94ab418a88 (patch) | |
| tree | 85778af30da4253a57d50018a9525d7750cf3c64 /reference_model/src/ops/tensor_ops.cc | |
| parent | 08965d35f728d93d8b215753b4b270a422fe39c9 (diff) | |
| download | reference_model-8ead6c48d878346dfadc7fb48ee9ec94ab418a88.tar.gz | |
[reference_model] tosa.fb name changes
This patch adjusts reference model for attribute name changes in
tosa.fb schema, and for obsoleted slice/tile/reshape attributes
also updated examples due to the breaking tosa flatbuffers changes
Signed-off-by: Tai Ly <tai.ly@arm.com>
Change-Id: I105eb99a4c35f289c5078aed0a7f9cbb6dfe9123
Diffstat (limited to 'reference_model/src/ops/tensor_ops.cc')
| -rw-r--r-- | reference_model/src/ops/tensor_ops.cc | 4 |
1 files changed, 2 insertions, 2 deletions
diff --git a/reference_model/src/ops/tensor_ops.cc b/reference_model/src/ops/tensor_ops.cc index 124dc87..609265c 100644 --- a/reference_model/src/ops/tensor_ops.cc +++ b/reference_model/src/ops/tensor_ops.cc @@ -496,11 +496,11 @@ int OpAvgPool2d<Dtype, AccDtype>::eval() LEVEL_CHECK(pad_left <= tosa_level.MAX_KERNEL, "pad_left should be smaller than or equal to MAX_KERNEL"); LEVEL_CHECK(pad_right <= tosa_level.MAX_KERNEL, "pad_right should be smaller than or equal to MAX_KERNEL"); - TOSA_REF_TYPE accum_dtype = ConvertDType(this->attribute->accum_dtype()); + TOSA_REF_TYPE accum_dtype = ConvertDType(this->attribute->acc_type()); DEBUG_INFO(OP, "perform AvgPool2d, input.shape=[%d,%d,%d,%d], output.shape=[%d,%d,%d,%d], kernel=[%d,%d], " - "stride=[%d,%d], pad=[%d,%d,%d,%d], accum_dtype=%s", + "stride=[%d,%d], pad=[%d,%d,%d,%d], acc_type=%s", in_batch, in_height, in_width, in_channels, out_batch, out_height, out_width, out_channels, kernel_y, kernel_x, stride_y, stride_x, pad_top, pad_bottom, pad_left, pad_right, EnumNamesDType()[accum_dtype]); |