diff options
Diffstat (limited to 'ethosu/vela/pass_packing.py')
-rw-r--r-- | ethosu/vela/pass_packing.py | 11 |
1 files changed, 5 insertions, 6 deletions
diff --git a/ethosu/vela/pass_packing.py b/ethosu/vela/pass_packing.py index 1ad5b4f7..9bf080e6 100644 --- a/ethosu/vela/pass_packing.py +++ b/ethosu/vela/pass_packing.py @@ -13,17 +13,16 @@ # WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied. # See the License for the specific language governing permissions and # limitations under the License. - - # Description: # Packs a subgraph with Neural Network Operations into Passes. Each Pass has one or more Operations. - -import enum import collections +import enum -from .nn_graph import Pass, PassPlacement +from .nn_graph import Pass +from .nn_graph import PassPlacement +from .operation import NpuBlockType +from .operation import Operation from .tensor import TensorPurpose -from .operation import Operation, NpuBlockType class PassFlags(enum.Flag): |