diff options
Diffstat (limited to 'ethosu/vela/operation.py')
| -rw-r--r-- | ethosu/vela/operation.py | 4 |
1 files changed, 4 insertions, 0 deletions
diff --git a/ethosu/vela/operation.py b/ethosu/vela/operation.py index 338f962e..e8a03b7d 100644 --- a/ethosu/vela/operation.py +++ b/ethosu/vela/operation.py @@ -194,6 +194,10 @@ input and output tensors, as well as an attribute dictionary.""" return inputs, axis + def get_dilation_h_w(self): + _, dilation_h, dilation_w, _ = self.attrs.get("dilation", (1, 1, 1, 1)) + return dilation_h, dilation_w + split_ops = set(("Split", "SplitV", "StridedSlice", "Slice", "UnpackReshaped")) def is_split_op(self): |