Searched refs:getResultAttrOfType (Results 1 – 4 of 4) sorted by relevance
384 AttrClass getResultAttrOfType(unsigned index, Identifier name) { in getResultAttrOfType() function388 AttrClass getResultAttrOfType(unsigned index, StringRef name) { in getResultAttrOfType() function
218 auto device_attr = func.getResultAttrOfType<StringAttr>( in PropagateDevicesToResults()
274 function.getResultAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) in GetReturnNode()
426 function.getResultAttrOfType<mlir::StringAttr>(i, kShardingAttr)) in ExtractShardingsFromFunction()