Home
last modified time | relevance | path

Searched refs:CPred (Results 1 – 25 of 57) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Doptimize.td29 CPred<"$0.getType().isa<RankedTensorType>()">,
30 CPred<"!$0.getType().cast<ShapedType>().isDynamicDim( "
34 CPred<"$0.getType().isa<RankedTensorType>()">,
35 CPred<"!$0.getType().cast<ShapedType>().getElementType().isa<ComplexType>()">
41 …Constraint<CPred<"TFL::IsBroadcastableElementsAttrs($0, $1) && TFL::IsDimensionsDegenerateExceptLa…
44 … CPred<"$_self.cast<ElementsAttr>().getType().getElementType().isF32()">, "float constant tensor">;
45 def DefinedByConv2D : Constraint<CPred<"llvm::isa_and_nonnull<mlir::TF::Conv2DOp>($0.getDefiningOp(…
47 def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>;
Dlegalize_hlo_patterns.td26 Constraint<CPred<"hlo::IsLegalNumpyRankedBroadcast($0, $1, $2)">,
33 def IsTFStyleBroadcast : Constraint<CPred<"IsTFStyleBroadcast($0, $1)">,
37 def IsNotTFStyleBroadcast : Constraint<Neg<CPred<"IsTFStyleBroadcast($0, $1)">>,
169 CPred<"$_self.cast<StringAttr>().getValue() == \"" # value # "\"">;
174 Or<[CPred<"!$_self">, HasCompareType<"FLOAT">, HasCompareType<"SIGNED">,
220 def IsZero : Constraint<CPred<
229 class FloatValueEquals<string val> : Constraint<CPred<
232 def SameValue : Constraint<CPred<"$0 == $1">>;
233 def FloatOrDefaultCompare : Constraint<CPred<
Dcanonicalize.td23 CPred<"getElementTypeOrSelf($0) == getElementTypeOrSelf($1)">>;
26 CPred<"$0.getType() == $1.getType()">>;
34 CPred<"llvm::all_of($0.getUsers(), [](mlir::OpOperand op) { "
/external/llvm-project/mlir/include/mlir/IR/
DOpBase.td60 // 1. CPred: the primitive leaf predicate.
83 // `CPred<"$_self.isa<BoolAttr>()">` (see the following sections for details).
90 class CPred<code pred> : Pred {
212 def IsVectorTypePred : CPred<"$_self.isa<::mlir::VectorType>()">;
215 def IsTensorTypePred : CPred<"$_self.isa<::mlir::TensorType>()">;
218 def IsMemRefTypePred : CPred<"$_self.isa<::mlir::MemRefType>()">;
222 : CPred<"$_self.isa<::mlir::UnrankedMemRefType>()">;
225 def IsShapedTypePred : CPred<"$_self.isa<::mlir::ShapedType>()">;
229 CPred<"$_self.cast<::mlir::ShapedType>().hasStaticShape()">;
232 def IsTupleTypePred : CPred<"$_self.isa<::mlir::TupleType>()">;
[all …]
/external/llvm-project/mlir/include/mlir/TableGen/
DPredicate.h68 class CPred : public Pred {
71 explicit CPred(const llvm::Record *record);
73 explicit CPred(const llvm::Init *init);
/external/llvm-project/mlir/include/mlir/Dialect/Async/IR/
DAsyncBase.td34 CPred<"$_self.isa<::mlir::async::TokenType>()">, "token type">,
45 CPred<"$_self.isa<::mlir::async::ValueType>()">,
60 CPred<"$_self.isa<::mlir::async::GroupType>()">, "group type">,
70 CPred<"$_self.isa<::mlir::async::ValueType>()">,
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dutils.td24 CPred<"tflite::NotFromQuantOpOrSameQuantType($0,$1)">>;
28 CPred<"getElementTypeOrSelf($0) == getElementTypeOrSelf($1)">>;
31 def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>;
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Doptimize_patterns.td26 …CPred<"$_self.isa<ElementsAttr>() && $_self.cast<ElementsAttr>().getType().getElementType().isF32(…
31 …CPred<"$_self.isa<ElementsAttr>() && $_self.cast<ElementsAttr>().getType().getElementType().isa<Fl…
35 def IsNoneType : Constraint<CPred<"$0.getType().isa<NoneType>()">>;
42 CPred<"$0.getType().cast<ShapedType>().getRank() <= " # n>>;
92 CPred<"TFL::CanFuseConvOrDepthwiseConv($0, $1, " # is_depthwise # ")">>;
250 CPred<"$0.getType().cast<ShapedType>().getRank() == " # n>>;
295 CPred<"$0.isa<DenseElementsAttr>() && "
300 def L2NormValidReduceIndex : Constraint<CPred<
360 def AreBroadcastableTypes : Constraint<CPred<
363 def OperandsBroadcastToOutputType : Constraint<CPred<
[all …]
Dlegalize_patterns.td24 CPred<"!$_self.isa<OpaqueElementsAttr>()">,
28 CPred<"$_self.cast<ElementsAttr>().getType().getElementType().isF32()">, "float constant tensor">;
31 …CPred<"$_self.cast<ElementsAttr>().getType().getElementType().isInteger(64)">, "Int 64 constant te…
61 def HasSameStaticShapesPred : CPred<"HasSameStaticShapes($0.getDefiningOp())">;
87 def IsIntList1XY1 : AttrConstraint<CPred<"TFIntListIs1XY1($_self)">>;
90 def IsAllOnes : AttrConstraint<CPred<"TFIntListIsAllOnes($_self)">>;
94 CPred<"$_self.cast<StringAttr>().getValue() == \"SAME\" || " #
419 CPred<"$_self.isa<DenseIntElementsAttr>() &&"
449 : CPred<"$_self.cast<DenseElementsAttr>().getType().getRank() == 0">;
452 : CPred<"$_self.cast<DenseElementsAttr>().getSplatValue()."
[all …]
Dprepare_patterns.td19 def FalseBoolAttr : AttrConstraint<CPred<"!$_self.getValue()">>;
22 CPred<"!$_self.isa<OpaqueElementsAttr>()">,
100 CPred<"llvm::isa<mlir::TFL::" # op # "Op>(*$0.getUsers().begin())">>;
/external/llvm-project/mlir/lib/TableGen/
DPredicate.cpp41 return static_cast<const CPred *>(this)->getConditionImpl(); in getCondition()
51 CPred::CPred(const llvm::Record *record) : Pred(record) { in CPred() function in CPred
56 CPred::CPred(const llvm::Init *init) : Pred(init) { in CPred() function in CPred
62 std::string CPred::getConditionImpl() const { in getConditionImpl()
/external/llvm-project/mlir/lib/Dialect/Shape/IR/
DShapeCanonicalization.td4 def AllInputShapesEq : Constraint<CPred< [{
10 def HasSingleElement : Constraint<CPred< [{
/external/llvm-project/mlir/include/mlir/Dialect/Shape/IR/
DShapeBase.td43 CPred<"$_self.isa<::mlir::shape::ComponentType>()">, "component type">,
52 CPred<"$_self.isa<::mlir::shape::ElementType>()">, "element type">,
61 CPred<"$_self.isa<::mlir::shape::ShapeType>()">, "shape">,
80 CPred<"$_self.isa<::mlir::shape::SizeType>()">, "size">,
93 CPred<"$_self.isa<::mlir::shape::ValueShapeType>()">, "value shape">,
130 CPred<"$_self.isa<::mlir::shape::WitnessType>()">, "witness">,
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/
DTosaTypesBase.td27 : Type<And<[CPred<"$_self.isa<mlir::quant::QuantizedType>()">,
28 CPred<"$_self.cast<mlir::quant::QuantizedType>()" #
130 CPred<"$_self.cast<::mlir::ArrayAttr>().size() <= " # n>,
148 CPred<"$_self.cast<StringAttr>().getValue() == \"BILINEAR\" || " #
/external/llvm-project/mlir/include/mlir/Dialect/LLVMIR/
DLLVMOpBase.td52 CPred<"$_self.isa<::mlir::LLVM::LLVMType>()">,
57 CPred<"$_self.isa<::mlir::LLVM::LLVMIntegerType>()">,
64 CPred<"$_self.cast<::mlir::LLVM::LLVMIntegerType>().getBitWidth() == "
79 CPred<"!$_self.isa<::mlir::LLVM::LLVMVoidType, "
85 CPred<"$_self.isa<::mlir::LLVM::LLVMBFloatType, "
94 def LLVM_AnyPointer : Type<CPred<"$_self.isa<::mlir::LLVM::LLVMPointerType>()">,
108 def LLVM_AnyStruct : Type<CPred<"$_self.isa<::mlir::LLVM::LLVMStructType>()">,
114 CPred<"$_self.cast<::mlir::LLVM::LLVMStructType>().isOpaque()">]>>;
124 CPred<"$_self.isa<::mlir::LLVM::LLVMStructType, "
134 def LLVM_AnyVector : Type<CPred<"$_self.isa<::mlir::LLVM::LLVMVectorType>()">,
/external/llvm-project/mlir/test/mlir-tblgen/
Dpredicate.td11 def I32OrF32 : Type<CPred<"$_self.isInteger(32) || $_self.isF32()">,
25 And<[CPred<"first">, CPred<"second">]>>]> {
Dexpect-symbol.td18 [(Constraint<CPred<"$0->getValue() == $1">> $attr, 42)]>;
/external/tensorflow/tensorflow/compiler/mlir/tfjs/transforms/
Doptimize_pattern.td24 def HasOneUse : Constraint<CPred<"$0.hasOneUse()">>;
28 CPred<"$0.getType().cast<ShapedType>().getRank() == "
/external/llvm-project/mlir/include/mlir/Dialect/Quant/
DQuantOpsBase.td35 Type<CPred<"$_self.isa<mlir::quant::QuantizedType>()">, "QuantizedType">;
65 CPred<"$_self.isa<UniformQuantizedType>()">,
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_utils.td57 CPred<"hlo::IsSequenceStartingWith0($_self)">>;
60 CPred<"!$_self.cast<ShapedType>().getElementType().isa<ComplexType>()">,
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_patterns.td36 def FalseBoolAttr : AttrConstraint<CPred<"!$_self.getValue()">>;
37 def TrueBoolAttr : AttrConstraint<CPred<"$_self.getValue()">>;
86 def AreBroadcastCompatible : Constraint<CPred<"AreBroadcastCompatible($0, $1)">,
252 : CPred<"$_self.cast<ElementsAttr>().getType().getNumElements() == 1">;
259 : Constraint<CPred<"(*$0.begin()).getType().isa<RankedTensorType>()">>;
262 : Constraint<CPred<"$0.getType().isa<RankedTensorType>()">>;
308 : Constraint<CPred<"CheckInnerDimStatic($0.getType().cast<ShapedType>(), &$_builder)">>;
537 def CanBeTranslatedToDynamicSlice : Constraint<CPred<
557 CPred<"ArgTypesMatchCallee($0[0].getOwner(), $1, $2)">>;
649 def BothElementTypesSameWidthIntOrFloat : Constraint<CPred<
[all …]
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_op_base.td95 CPred<"mlir::TF::BroadcastCompatible("
101 CPred<"TF::AreCastCompatible(llvm::makeArrayRef({" #
116 CPred<"$_op.getOperand(" # n # ").getType().isa<UnrankedTensorType>()">;
119 CPred<"$_op.getResult(" # n # ").getType().isa<UnrankedTensorType>()">;
125 CPred<"$_op.getOperand(" # n #
132 CPred<"$_op.getResult(" # n #
200 Attr<CPred<"$_self.isa<mlir::TF::" # name # "Attr>()">,
220 Type<CPred<"$_self.isa<mlir::TF::TensorFlowType>()">, "TensorFlow type">;
224 Type<CPred<"$_self.isa<mlir::TF::" # name # "Type>()">,
435 CPred<!foldl(
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td32 def TFL_Str : Type<CPred<"$_self.isa<mlir::TF::StringType>()">,
39 def TFL_Quint8 : Type<CPred<"$_self.isa<mlir::TF::Quint8Type>()">,
90 CPred<"$_self.isa<OpaqueElementsAttr>() ">,
91 CPred<"$_self.cast<OpaqueElementsAttr>().getType()"
119 CPred<"TFL::VerifyOperandsHaveSameShapesOrBroadcastableShape("
184 CPred<"$_op.getOperand(" # i # ").getType().isa<NoneType>()">;
187 CPred<"$_op.getOperand(" # n # ").getType().isa<UnrankedTensorType>()">;
195 CPred<"$_op.getOperand(" # n #
200 CPred<"$_op.getOperand(" # n # ").getType().isa<RankedTensorType>()">,
201 CPred<"$_op.getOperand(" # n # ").getType().cast<ShapedType>().getRank() == "
[all …]
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization.td31 : Type<And<[CPred<"$_self.isa<mlir::quant::QuantizedType>()">,
32 CPred<"$_self.cast<mlir::quant::QuantizedType>()" #
/external/llvm-project/flang/include/flang/Optimizer/Dialect/
DFIROps.td29 def fir_Type : Type<CPred<"fir::isa_fir_or_std_type($_self)">,
33 def fir_CharacterType : Type<CPred<"$_self.isa<fir::CharacterType>()">,
35 def fir_ComplexType : Type<CPred<"$_self.isa<fir::CplxType>()">,
37 def fir_IntegerType : Type<CPred<"$_self.isa<fir::IntType>()">,
39 def fir_LogicalType : Type<CPred<"$_self.isa<fir::LogicalType>()">,
41 def fir_RealType : Type<CPred<"$_self.isa<fir::RealType>()">,
54 def fir_RecordType : Type<CPred<"$_self.isa<fir::RecordType>()">,
58 def fir_SequenceType : Type<CPred<"$_self.isa<fir::SequenceType>()">,
67 def fir_ReferenceType : Type<CPred<"$_self.isa<fir::ReferenceType>()">,
71 def fir_HeapType : Type<CPred<"$_self.isa<fir::HeapType>()">,
[all …]

123