Searched refs:check_layout (Results 1 – 4 of 4) sorted by relevance
65 def check_layout(self, layout, file_size): member in LayoutScraper125 self.check_layout(layout_data, self.os_if.get_file_size(file_name))
22 from tensorflow.dtensor.python.api import check_layout
357 def check_layout(tensor: ops.Tensor, layout: layout_lib.Layout) -> None: function
40 name: "check_layout"