Skip to content

Commit

Permalink
[converter&docs] fix misc & converter tests (#283)
Browse files Browse the repository at this point in the history
* fix misc tests

* debug

* fix constant list unpacking
  • Loading branch information
peterjc123 authored Mar 24, 2024
1 parent 655dd32 commit 663e7ce
Show file tree
Hide file tree
Showing 2 changed files with 6 additions and 1 deletion.
2 changes: 2 additions & 0 deletions docs/op_matrix.md
Original file line number Diff line number Diff line change
Expand Up @@ -94,6 +94,8 @@ Operators that are implemented in Python
| `aten::hardtanh_` | |
| `aten::im2col` | only 4-D input tensors (batched image-like tensors) are supported |
| `aten::index` | Multiple indices for aten::index is not supported |
| `aten::index_put` | aten::index_put_ with accumulate=True is not supported |
| `aten::index_put_` | aten::index_put_ with accumulate=True is not supported |
| `aten::index_select` | |
| `aten::instance_norm` | |
| `aten::layer_norm` | |
Expand Down
5 changes: 4 additions & 1 deletion tinynn/converter/operators/torch/aten.py
Original file line number Diff line number Diff line change
Expand Up @@ -3077,7 +3077,10 @@ def parse(self, node, attrs, args, graph_converter):
input_names, self.input_tensors[1], graph_converter=graph_converter, non_existent_as_buffer=True
)
else:
indices_tensors = [self.find_or_create_input(1, graph_converter)]
if type(self.input_tensors[1]) in (tuple, list):
indices_tensors = [self.create_attr_tensor(x) for x in self.input_tensors[1]]
else:
indices_tensors = [self.find_or_create_input(1, graph_converter)]

dim = input_tensor.tensor.ndim

Expand Down

0 comments on commit 663e7ce

Please sign in to comment.