Searched defs:layout_attr (Results 1 – 7 of 7) sorted by relevance
89 const std::string layout_attr = copy_to_mesh.layout().str(); in CloneOpToCluster() local196 const std::string layout_attr = copy_to_mesh.layout().str(); in LowerToSendRecv() local
92 auto layout_attr = function.getArgAttrOfType<mlir::StringAttr>( in PropagateFunctionArgAttrToLayoutOp() local
199 auto layout_attr = func_op.getArgAttrOfType<mlir::StringAttr>( in ExtractLayoutFromOperand() local
88 mlir::ArrayAttr layout_attr, mlir::DenseIntElementsAttr indices_attr2, in MergeAttributes()
148 mlir::StringRef layout_attr = relayout.layout(); in ComputeRelayoutLayout() local176 mlir::StringRef layout_attr = relayout.layout(); in ExpandOp() local
149 if (auto layout_attr = attr.dyn_cast<LayoutAttr>()) in printAttribute() local
191 absl::string_view layout_attr, in ParseAttrMap()