Searched refs:targetLayout (Results 1 – 1 of 1) sorted by relevance
826 def __init__(self, targetLayout="nchw", name=None): argument828 self.targetLayout = targetLayout.lower()829 assert DataLayoutConverter.IsCompatible(self.targetLayout)830 self.perm = (0, 3, 1, 2) if self.targetLayout == "nchw" else (0, 2, 3, 1)831 self.param = True if self.targetLayout == "nchw" else False838 self.name = self.targetLayout