Searched defs:fetch_layout (Results 1 – 3 of 3) sorted by relevance
341 def fetch_layout(tensor: ops.Tensor) -> layout_lib.Layout: function
292 def fetch_layout(self, dtensor: Any) -> layout_lib.Layout: member in DTensorDevice
327 fetch_layout(struct rxkb_context *ctx, const char *layout, const char *variant) in fetch_layout() function