diff options
Diffstat (limited to 'ethosu/vela/mark_tensors.py')
| -rw-r--r-- | ethosu/vela/mark_tensors.py | 6 |
1 files changed, 2 insertions, 4 deletions
diff --git a/ethosu/vela/mark_tensors.py b/ethosu/vela/mark_tensors.py index c42a28df..36508762 100644 --- a/ethosu/vela/mark_tensors.py +++ b/ethosu/vela/mark_tensors.py @@ -13,16 +13,14 @@ # 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: # Mark purpose and select formats for Tensors. Also compresses the weights. - from . import rewrite_graph from . import weight_compressor from .architecture_features import Block -from .tensor import TensorPurpose, TensorFormat from .operation import NpuBlockType +from .tensor import TensorFormat +from .tensor import TensorPurpose def purpose_from_list(lst): |