| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.h | 69 static mlir::NamedAttribute ConvertSourceTargetPairs(
76 static mlir::NamedAttribute ConvertReplicaGroups(
134 mlir::NamedAttribute ConvertComparisonDirection(
138 mlir::NamedAttribute ConvertComparisonType(Comparison::Type type);
149 mlir::NamedAttribute ConvertPadding(llvm::ArrayRef<int64_t> padding);
152 mlir::NamedAttribute ConvertChannelHandle(
156 mlir::NamedAttribute ConvertChannelHandle(const xla::ChannelHandle& channel);
|
| D | operator_writer_gen.cc | 37 using mlir::tblgen::NamedAttribute;
42 const mlir::tblgen::NamedAttribute& named_attr) { in GetDefaultAttrExport()
111 auto named_attr = arg.get<NamedAttribute*>(); in BuildOperator()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | OperationSupport.h | 222 using const_iterator = SmallVectorImpl<NamedAttribute>::const_iterator;
223 using const_reference = const NamedAttribute &;
224 using reference = NamedAttribute &;
228 NamedAttrList(ArrayRef<NamedAttribute> attributes);
245 void append(ArrayRef<NamedAttribute> newAttributes);
254 void assign(ArrayRef<NamedAttribute> range) { in assign()
263 void push_back(NamedAttribute newAttribute);
270 Optional<NamedAttribute> findDuplicate() const;
277 ArrayRef<NamedAttribute> getAttrs() const;
284 Optional<NamedAttribute> getNamed(StringRef name) const;
[all …]
|
| D | Operation.h | 36 ArrayRef<NamedAttribute> attributes,
307 ArrayRef<NamedAttribute> getAttrs() { return attrs.getAttrs(); } in getAttrs()
361 : public llvm::filter_iterator<ArrayRef<NamedAttribute>::iterator,
362 bool (*)(NamedAttribute)> {
363 static bool filter(NamedAttribute attr) { in filter()
368 explicit dialect_attr_iterator(ArrayRef<NamedAttribute>::iterator it, in dialect_attr_iterator()
369 ArrayRef<NamedAttribute>::iterator end) in dialect_attr_iterator()
370 : llvm::filter_iterator<ArrayRef<NamedAttribute>::iterator, in dialect_attr_iterator()
371 bool (*)(NamedAttribute)>(it, end, &filter) {} in dialect_attr_iterator()
396 SmallVector<NamedAttribute, 16> attrs; in setDialectAttrs()
|
| D | Attributes.h | 130 using NamedAttribute = std::pair<Identifier, Attribute>; variable
132 bool operator<(const NamedAttribute &lhs, const NamedAttribute &rhs);
133 bool operator<(const NamedAttribute &lhs, StringRef rhs);
|
| D | FunctionSupport.h | 64 inline ArrayRef<NamedAttribute> getArgAttrs(Operation *op, unsigned index) { in getArgAttrs()
70 inline ArrayRef<NamedAttribute> getResultAttrs(Operation *op, unsigned index) { in getResultAttrs()
299 ArrayRef<NamedAttribute> getArgAttrs(unsigned index) { in getArgAttrs()
330 void setArgAttrs(unsigned index, ArrayRef<NamedAttribute> attributes);
362 ArrayRef<NamedAttribute> getResultAttrs(unsigned index) { in getResultAttrs()
393 void setResultAttrs(unsigned index, ArrayRef<NamedAttribute> attributes);
562 unsigned index, ArrayRef<NamedAttribute> attributes) { in setArgAttrs()
623 unsigned index, ArrayRef<NamedAttribute> attributes) { in setResultAttrs()
|
| D | BuiltinAttributes.h | 123 using ValueType = ArrayRef<NamedAttribute>;
129 static DictionaryAttr get(ArrayRef<NamedAttribute> value,
134 static DictionaryAttr getWithSorted(ArrayRef<NamedAttribute> value,
137 ArrayRef<NamedAttribute> getValue() const;
144 Optional<NamedAttribute> getNamed(StringRef name) const;
145 Optional<NamedAttribute> getNamed(Identifier name) const;
148 using iterator = llvm::ArrayRef<NamedAttribute>::iterator;
157 static bool sort(ArrayRef<NamedAttribute> values,
158 SmallVectorImpl<NamedAttribute> &storage);
164 static bool sortInPlace(SmallVectorImpl<NamedAttribute> &array);
[all …]
|
| D | Dialect.h | 108 NamedAttribute);
117 NamedAttribute);
121 virtual LogicalResult verifyOperationAttribute(Operation *, NamedAttribute) { in verifyOperationAttribute() argument
|
| /external/llvm-project/mlir/unittests/TableGen/ |
| D | StructsGenTest.cpp | 61 llvm::SmallVector<mlir::NamedAttribute, 5> newValues(expectedValues.begin(), in TEST()
66 auto wrongAttr = mlir::NamedAttribute(wrongId, expectedValues[0].second); in TEST()
83 llvm::SmallVector<mlir::NamedAttribute, 4> newValues( in TEST()
88 auto wrongAttr = mlir::NamedAttribute(wrongId, expectedValues[0].second); in TEST()
104 llvm::SmallVector<mlir::NamedAttribute, 4> newValues( in TEST()
113 auto wrongAttr = mlir::NamedAttribute(id, elementsAttr); in TEST()
129 llvm::SmallVector<mlir::NamedAttribute, 3> newValues( in TEST()
|
| D | OpBuildGen.cpp | 52 std::vector<NamedAttribute> attrs) { in verifyOp()
102 ArrayRef<NamedAttribute> noAttrs;
103 std::vector<NamedAttribute> attrStorage;
104 ArrayRef<NamedAttribute> attrs;
|
| /external/llvm-project/mlir/lib/IR/ |
| D | OperationSupport.cpp | 25 NamedAttrList::NamedAttrList(ArrayRef<NamedAttribute> attributes) { in NamedAttrList()
33 ArrayRef<NamedAttribute> NamedAttrList::getAttrs() const { return attrs; } in getAttrs()
35 Optional<NamedAttribute> NamedAttrList::findDuplicate() const { in findDuplicate()
36 Optional<NamedAttribute> duplicate = in findDuplicate()
72 void NamedAttrList::append(ArrayRef<NamedAttribute> newAttributes) { in append()
86 DictionaryAttr::sort(ArrayRef<NamedAttribute>{in_start, in_end}, attrs); in assign()
90 void NamedAttrList::push_back(NamedAttribute newAttribute) { in push_back()
102 static auto *findAttr(SmallVectorImpl<NamedAttribute> &attrs, T name, in findAttr()
106 attrs, [name](NamedAttribute attr) { return attr.first == name; }); in findAttr()
128 Optional<NamedAttribute> NamedAttrList::getNamed(StringRef name) const { in getNamed()
[all …]
|
| D | BuiltinAttributes.cpp | 58 static bool dictionaryAttrSort(ArrayRef<NamedAttribute> value, in dictionaryAttrSort()
59 SmallVectorImpl<NamedAttribute> &storage) { in dictionaryAttrSort()
99 static Optional<NamedAttribute>
100 findDuplicateElement(ArrayRef<NamedAttribute> value) { in findDuplicateElement()
101 const Optional<NamedAttribute> none{llvm::None}; in findDuplicateElement()
110 [](NamedAttribute l, NamedAttribute r) { return l.first == r.first; }); in findDuplicateElement()
114 bool DictionaryAttr::sort(ArrayRef<NamedAttribute> value, in sort()
115 SmallVectorImpl<NamedAttribute> &storage) { in sort()
122 bool DictionaryAttr::sortInPlace(SmallVectorImpl<NamedAttribute> &array) { in sortInPlace()
129 Optional<NamedAttribute>
[all …]
|
| D | Attributes.cpp | 49 bool mlir::operator<(const NamedAttribute &lhs, const NamedAttribute &rhs) { in operator <()
52 bool mlir::operator<(const NamedAttribute &lhs, StringRef rhs) { in operator <()
|
| D | AttributeDetail.h | 72 NamedAttribute> {
73 using KeyTy = ArrayRef<NamedAttribute>;
77 static DictionaryAttributeStorage *get(ArrayRef<NamedAttribute> attrs);
85 auto size = DictionaryAttributeStorage::totalSizeToAlloc<NamedAttribute>( in construct()
92 result->getTrailingObjects<NamedAttribute>()); in construct()
97 ArrayRef<NamedAttribute> getElements() const { in getElements()
98 return {getTrailingObjects<NamedAttribute>(), numElements}; in getElements()
103 NamedAttribute>;
106 size_t numTrailingObjects(OverloadToken<NamedAttribute>) const { in numTrailingObjects()
|
| D | BuiltinDialect.cpp | 75 ArrayRef<NamedAttribute> attrs) { in create()
83 SmallVector<NamedAttribute, 8> attrRef(attrs); in create()
87 ArrayRef<NamedAttribute> attrs, in create()
95 FunctionType type, ArrayRef<NamedAttribute> attrs, in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | attribute_utils.h | 28 for (const NamedAttribute &attr : from->getAttrs()) in CopyAttributes()
34 CopyAttributes(from, to, [](const NamedAttribute &attr) { in CopyUnderscoredAttributes()
45 CopyAttributes(from, to, [&device](const NamedAttribute &attr) { in CopyDeviceAndUnderscoredAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_saved_model.h | 31 NamedAttribute named_attr) override;
35 NamedAttribute named_attr) override;
37 NamedAttribute named_attr) override;
|
| D | tf_op_interfaces.h | 41 ArrayRef<NamedAttribute> additional_attributes = {})
56 SmallVector<NamedAttribute, 4> additional_attributes;
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | contraction_fusion.cc | 93 SmallVector<NamedAttribute, 4> attrs(fuse_into->getAttrs().begin(), in matchAndRewrite()
105 auto is_fusion = [&](const NamedAttribute &attr) -> bool { in matchAndRewrite()
110 NamedAttribute fusion_attr(fusion_id, ArrayAttr::get(ctx, {fusion_name})); in matchAndRewrite()
122 attrs.push_back(NamedAttribute(fusion_id, ArrayAttr::get(ctx, updated))); in matchAndRewrite()
|
| D | fused_kernel_matcher.cc | 159 std::vector<NamedAttribute> attrs = contraction.getAttrs(); in matchAndRewrite()
162 NamedAttribute(Identifier::get("fused_ops", context), fused_ops_attr)); in matchAndRewrite()
167 NamedAttribute(Identifier::get("epsilon", context), epsilon)); in matchAndRewrite()
|
| /external/llvm-project/mlir/include/mlir/TableGen/ |
| D | Argument.h | 37 struct NamedAttribute { struct
60 using Argument = llvm::PointerUnion<NamedAttribute *, NamedTypeConstraint *>; argument
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.h | 57 llvm::SmallVectorImpl<mlir::NamedAttribute> &attributes);
72 Location loc, llvm::SmallVectorImpl<mlir::NamedAttribute> *attributes);
|
| /external/llvm-project/flang/include/flang/Optimizer/Dialect/ |
| D | FIROpsSupport.h | 55 llvm::ArrayRef<mlir::NamedAttribute> attrs = {});
60 llvm::ArrayRef<mlir::NamedAttribute> attrs = {});
|
| /external/llvm-project/mlir/test/mlir-tblgen/ |
| D | op-decl.td | 90 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes, u…
188 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
200 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
212 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
226 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
236 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
238 …ate &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
249 …e resultTypes, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
250 …ate &odsState, ::mlir::ValueRange operands, ::llvm::ArrayRef<::mlir::NamedAttribute> attributes = …
|
| /external/llvm-project/flang/lib/Lower/ |
| D | FIRBuilder.cpp | 81 llvm::SmallVector<mlir::NamedAttribute, 2> attrs; in allocateLocal()
178 mlir::NamedAttribute dataAttr(valTag, strAttr); in createStringLit()
180 mlir::NamedAttribute sizeAttr(sizeTag, getI64IntegerAttr(data.size())); in createStringLit()
181 llvm::SmallVector<mlir::NamedAttribute, 2> attrs{dataAttr, sizeAttr}; in createStringLit()
|