Searched defs:source_layout (Results 1 – 3 of 3) sorted by relevance
72 LayoutProto source_layout = 1; field
182 def copy_to_mesh(self, tensor, new_layout, source_layout=None) -> ops.Tensor: argument
1736 xla::LayoutProto source_layout = in ExportXlaOp() local