Searched refs:torch_layout_to_ck_layout (Results 1 – 1 of 1) sorted by relevance
28 def torch_layout_to_ck_layout(torch_layout): function147 if op.a_layout != torch_layout_to_ck_layout(X_meta):149 if op.b_layout != torch_layout_to_ck_layout(W_meta):151 if op.c_layout != torch_layout_to_ck_layout(Y_meta):267 op.ds_layouts = (torch_layout_to_ck_layout(Bias.get_layout()),)336 torch_layout_to_ck_layout(m) for m in (X_meta, W_meta, Y_meta)