Home
last modified time | relevance | path

Searched refs:AttrConstraint (Results 1 – 25 of 26) sorted by relevance

12

/external/tensorflow/tensorflow/core/framework/
Dkernel_def_builder.cc37 KernelDefBuilder& KernelDefBuilder::AttrConstraint<int64>( in AttrConstraint() function in tensorflow::KernelDefBuilder
50 KernelDefBuilder& KernelDefBuilder::AttrConstraint<int64>(const char* attr_name, in AttrConstraint() function in tensorflow::KernelDefBuilder
52 return AttrConstraint( in AttrConstraint()
58 KernelDefBuilder& KernelDefBuilder::AttrConstraint<string>( in AttrConstraint() function in tensorflow::KernelDefBuilder
70 KernelDefBuilder& KernelDefBuilder::AttrConstraint<string>( in AttrConstraint() function in tensorflow::KernelDefBuilder
72 return AttrConstraint( in AttrConstraint()
78 KernelDefBuilder& KernelDefBuilder::AttrConstraint<const char*>( in AttrConstraint() function in tensorflow::KernelDefBuilder
90 KernelDefBuilder& KernelDefBuilder::AttrConstraint<const char*>( in AttrConstraint() function in tensorflow::KernelDefBuilder
92 return AttrConstraint(attr_name, in AttrConstraint()
98 KernelDefBuilder& KernelDefBuilder::AttrConstraint<bool>(const char* attr_name, in AttrConstraint() function in tensorflow::KernelDefBuilder
Dkernel_def_builder_test.cc79 .AttrConstraint("T", int64{5}) in TEST()
96 .AttrConstraint("U", gtl::ArraySlice<int64>{int64{5}, int64{17}}) in TEST()
97 .AttrConstraint("V", string("proto")) in TEST()
120 .AttrConstraint("T", "hi") in TEST()
137 .AttrConstraint("U", gtl::ArraySlice<const char*>{"boo", "ya"}) in TEST()
138 .AttrConstraint("V", string("proto")) in TEST()
Dkernel_def_builder.h46 KernelDefBuilder& AttrConstraint(const char* attr_name,
51 KernelDefBuilder& AttrConstraint(const char* attr_name, T allowed);
Dkernel_def.proto20 message AttrConstraint { message
28 repeated AttrConstraint constraint = 3;
/external/python/pyasn1-modules/pyasn1_modules/
Drfc6010.py34 class AttrConstraint(univ.Sequence): class
37 AttrConstraint.componentType = namedtype.NamedTypes(
47 AttrConstraintList.componentType = AttrConstraint()
/external/llvm-project/mlir/include/mlir/TableGen/
DAttribute.h33 class AttrConstraint : public Constraint {
35 explicit AttrConstraint(const llvm::Record *record);
47 class Attribute : public AttrConstraint {
/external/llvm-project/mlir/lib/TableGen/
DAttribute.cpp34 AttrConstraint::AttrConstraint(const Record *record) in AttrConstraint() function in AttrConstraint
40 bool AttrConstraint::isSubClassOf(StringRef className) const { in isSubClassOf()
44 Attribute::Attribute(const Record *record) : AttrConstraint(record) { in Attribute()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dtf2xla_supported_ops.cc50 for (const KernelDef::AttrConstraint& constraint : kdef->constraint()) { in PrintSupportedOps()
Dxla_op_registry.cc275 KernelDef::AttrConstraint* attr_constraint = kdef->add_constraint(); in RegisterCompilationKernels()
Dtf2xla_util.cc529 for (KernelDef::AttrConstraint& constraint : *kdef->mutable_constraint()) { in AddDtypeToKernelDefConstraint()
/external/tensorflow/tensorflow/compiler/mlir/hlo/include/mlir-hlo/Dialect/mhlo/IR/
Dhlo_utils.td56 def IdentityBroadcastDims : AttrConstraint<
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_patterns.td87 def IsIntList1XY1 : AttrConstraint<CPred<"TFIntListIs1XY1($_self)">>;
90 def IsAllOnes : AttrConstraint<CPred<"TFIntListIsAllOnes($_self)">>;
93 def IsSameOrValid : AttrConstraint<
481 : AttrConstraint<CPred<"$_self.cast<::mlir::IntegerAttr>().getInt() == 0">>;
Dprepare_patterns.td19 def FalseBoolAttr : AttrConstraint<CPred<"!$_self.getValue()">>;
/external/llvm-project/mlir/include/mlir/IR/
DOpBase.td182 class AttrConstraint<Pred predicate, string description = ""> :
199 // * Use Attr (a subclass for AttrConstraint) for
201 // * Use AttrConstraint to specify
750 AttrConstraint<condition, descr> {
1549 class ConstantAttr<Attr attribute, string val> : AttrConstraint<
1568 class Confined<Attr attr, list<AttrConstraint> constraints> : Attr<
1584 // An AttrConstraint that holds if all attr constraints specified in
1586 class AllAttrConstraintsOf<list<AttrConstraint> constraints> : AttrConstraint<
1592 class IntMinValue<int n> : AttrConstraint<
1596 class IntMaxValue<int n> : AttrConstraint<
[all …]
/external/tensorflow/tensorflow/core/kernels/
Ddepthtospace_op.cc191 .AttrConstraint("data_format", "NHWC"), \
Dspacetodepth_op.cc207 .AttrConstraint("data_format", "NHWC"), \
Dconv_grad_filter_ops.cc618 .AttrConstraint("data_format", "NHWC"), \
624 .AttrConstraint("data_format", "NHWC"), \
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/IR/
DTosaTypesBase.td129 class ArrayMaxCt<int n> : AttrConstraint<
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dlegalize_hlo_patterns.td173 def IsTFCompareType : AttrConstraint<
Dlower_tf.td38 def TrueBoolAttr : AttrConstraint<CPred<"$_self.getValue()">>;
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dlegalize_tf_patterns.td36 def FalseBoolAttr : AttrConstraint<CPred<"!$_self.getValue()">>;
37 def TrueBoolAttr : AttrConstraint<CPred<"$_self.getValue()">>;
/external/llvm-project/mlir/docs/
DOpDefinitions.md98 * The `AttrConstraint` class hierarchy: They are used to specify the
200 `AttrConstraint` class hierarchy. See [Constraints](#constraints) for more
248 To declare an optional attribute, wrap the `AttrConstraint` for the attribute
253 To declare an attribute with a default value, wrap the `AttrConstraint` for the
1099 Similarly, a set of `AttrConstraint`s are created for helping modelling
1210 def HasSomeProperty : AttrConstraint<CPred<"HasSomeProperty($_self)">,
/external/llvm-project/mlir/test/lib/Dialect/Test/
DTestOps.td803 def Constraint0 : AttrConstraint<
807 def Constraint1 : AttrConstraint<
960 def MoreConstraint : AttrConstraint<
/external/llvm-project/mlir/tools/mlir-tblgen/
DRewriterGen.cpp541 } else if (isa<AttrConstraint>(constraint)) { in emitMatchLogic()
/external/tensorflow/tensorflow/compiler/mlir/lite/ir/
Dtfl_ops.td336 def TFL_FloatNonNegative : AttrConstraint<
341 def TFL_BoolTrue : AttrConstraint<
345 def TFL_BoolFalse : AttrConstraint<
349 class TFL_StringEqualsTo<string value> : AttrConstraint<
354 class TFL_ArrayMaxCount<int n> : AttrConstraint<
359 class TFL_IntEqualsTo<int n> : AttrConstraint<

12