Searched defs:desired_layout (Results 1 – 4 of 4) sorted by relevance
52 const Layout desired_layout = all_scatter.output_layout(); in GetScatterGroupAssignment() local88 const Layout desired_layout = all_scatter.output_layout(); in ApplyOptimization() local
107 const Layout& original_layout, const Layout& desired_layout, in EmitAllScatter()
231 const Layout& desired_layout) { in ComputeOneHot()
632 const Layout desired_layout = all_scatter.output_layout(); in LowerAllScatterOp() local