Searched defs:to_mkldnn (Results 1 – 2 of 2) sorted by relevance
206 def to_mkldnn(module, dtype=torch.float): function
726 auto to_mkldnn = in ComputeSubgraphInMKLDNN() local