Searched refs:targetLayout (Results 1 – 1 of 1) sorted by relevance
865 def __init__(self, targetLayout="nchw", name=None): argument867 self.targetLayout = targetLayout.lower()868 assert DataLayoutConverter.IsCompatible(self.targetLayout)869 self.perm = (0, 3, 1, 2) if self.targetLayout == "nchw" else (0, 2, 3, 1)870 self.param = True if self.targetLayout == "nchw" else False877 self.name = self.targetLayout