diff options
| author | Kevin Cheng <kevin.cheng@arm.com> | 2021-10-14 17:09:57 -0700 |
|---|---|---|
| committer | Kevin Cheng <kevin.cheng@arm.com> | 2021-10-18 18:50:08 +0000 |
| commit | cc61be36c3b0f5cd1ea719e129a54fd48a6ee9a2 (patch) | |
| tree | 2d664f87e3fdd75de8c6794f6f6c8d6364ece6bb /reference_model/src/ops/data_layout.h | |
| parent | e807aae606a78d923a2565052f7c2179e3050650 (diff) | |
| download | reference_model-cc61be36c3b0f5cd1ea719e129a54fd48a6ee9a2.tar.gz | |
More ERROR_IF supports
- Also delay tensor allocation after operator being validated
ERROR_IF can be caught first before 0 or negative dimension set the graph_status to UNPREDICTABLE
- Rescale, Argmax, FullyConnected, Matmul, Pad, Reshape, Slice, Transpose, Clamp, Concat, Equal, Greater, GreaterEqual, Table
Signed-off-by: Kevin Cheng <kevin.cheng@arm.com>
Change-Id: I4e1b3e5794fe195ce1a37e28443ae584645a3b91
Diffstat (limited to 'reference_model/src/ops/data_layout.h')
| -rw-r--r-- | reference_model/src/ops/data_layout.h | 1 |
1 files changed, 1 insertions, 0 deletions
diff --git a/reference_model/src/ops/data_layout.h b/reference_model/src/ops/data_layout.h index c9c2602..9f44fc7 100644 --- a/reference_model/src/ops/data_layout.h +++ b/reference_model/src/ops/data_layout.h @@ -63,6 +63,7 @@ protected: Eigen::array<std::pair<ptrdiff_t, ptrdiff_t>, Rank> paddings_array; TosaReference::TensorTemplate<TIn>* in; TosaReference::TensorTemplate<TOut>* out; + TosaReference::TensorTemplate<Eigen::Tensor<int32_t, 2>>* paddings; TosaPadQuantInfo* qinfo; }; |