Searched defs:get_memory_format (Results 1 – 2 of 2) sorted by relevance
34 def get_memory_format(dim_order: Optional[List[int]]) -> torch.memory_format: function
28 Optional<MemoryFormat> get_memory_format(OptionalArrayRef<int64_t> dim_order) { in get_memory_format() function