Searched refs:targetLayout (Results 1 – 1 of 1) sorted by relevance
810 def __init__(self, targetLayout="nchw", name=None): argument812 self.targetLayout = targetLayout.lower()813 assert DataLayoutConverter.IsCompatible(self.targetLayout)814 self.perm = (0, 3, 1, 2) if self.targetLayout == "nchw" else (0, 2, 3, 1)815 self.param = True if self.targetLayout == "nchw" else False822 self.name = self.targetLayout