Home
last modified time | relevance | path

Searched refs:TypeAttr (Results 1 – 25 of 80) sorted by relevance

1234

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconvert_type.h42 mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr);
46 bool NotFromQuantOpOrSameQuantType(mlir::Value val, mlir::TypeAttr qtype_attr);
Dlstm_utils.cc421 /*input_to_input_intermediate=*/mlir::TypeAttr(), in RewriteFunc()
422 /*input_to_forget_intermediate=*/mlir::TypeAttr(), in RewriteFunc()
423 /*input_to_cell_intermediate=*/mlir::TypeAttr(), in RewriteFunc()
424 /*input_to_output_intermediate=*/mlir::TypeAttr(), in RewriteFunc()
425 /*effective_hidden_scale_intermediate=*/mlir::TypeAttr()); in RewriteFunc()
749 /*input_to_input_intermediate=*/mlir::TypeAttr(), in ConvertKerasLSTMLayer()
750 /*input_to_forget_intermediate=*/mlir::TypeAttr(), in ConvertKerasLSTMLayer()
751 /*input_to_cell_intermediate=*/mlir::TypeAttr(), in ConvertKerasLSTMLayer()
752 /*input_to_output_intermediate=*/mlir::TypeAttr(), in ConvertKerasLSTMLayer()
753 /*effective_hidden_scale_intermediate=*/mlir::TypeAttr()); in ConvertKerasLSTMLayer()
Dconvert_type.cc147 mlir::Type GetShapeStrippedType(mlir::TypeAttr type_attr) { in GetShapeStrippedType()
157 bool NotFromQuantOpOrSameQuantType(mlir::Value val, mlir::TypeAttr qtype_attr) { in NotFromQuantOpOrSameQuantType()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/c/
Dc_api_unified_experimental_mlir.cc310 attrs_[attr_name] = TypeAttr::get(mlir_type); in SetAttrType()
371 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create()
395 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create()
414 TypeAttr type_attr = attr.dyn_cast<TypeAttr>(); in Create()
600 attrs_[arg_def.type_attr()] = TypeAttr::get(expected_type); in AddInput()
643 TypeAttr::get(cast<MlirTensor>(inputs.front())->getElementType()); in AddInputList()
650 types.push_back(TypeAttr::get(cast<MlirTensor>(input)->getElementType())); in AddInputList()
/external/llvm-project/mlir/include/mlir/Dialect/SPIRV/
DTargetAndABI.td47 StructFieldAttr<"a_type", TypeAttr>,
48 StructFieldAttr<"b_type", TypeAttr>,
49 StructFieldAttr<"c_type", TypeAttr>,
50 StructFieldAttr<"result_type", TypeAttr>,
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
Dquantization_context.cc133 input_specs.push_back(TypeAttr::get(requantize.params)); in Finalize()
135 input_specs.push_back(TypeAttr::get(state.params)); in Finalize()
148 output_specs.push_back(TypeAttr::get(requantize.params)); in Finalize()
150 output_specs.push_back(TypeAttr::get(state.params)); in Finalize()
289 params_attr.cast<TypeAttr>().getValue().dyn_cast<QuantParams>(); in InitializeState()
Ddevice_target.cc110 input_specs.push_back(spec.cast<TypeAttr>().getValue()); in DecomposeMultiplyAccumulateScale()
113 out_specs.push_back(spec.cast<TypeAttr>().getValue()); in DecomposeMultiplyAccumulateScale()
163 .cast<TypeAttr>() in DecomposeSameScale()
Dquantization_utils.cc124 TypeAttr RescaleQuantizedType(Type input, Attribute factor) { in RescaleQuantizedType()
148 return TypeAttr::get(new_type); in RescaleQuantizedType()
155 TypeAttr GetQuantizedTypeAttr(Builder builder, Type input_type, Attribute min, in GetQuantizedTypeAttr()
185 return TypeAttr::get(final_type); in GetQuantizedTypeAttr()
251 TypeAttr CastQuantizedTypeAttrFromExpressedType(Builder builder, in CastQuantizedTypeAttrFromExpressedType()
252 TypeAttr source, Type target, in CastQuantizedTypeAttrFromExpressedType()
268 return TypeAttr::get(final_type); in CastQuantizedTypeAttrFromExpressedType()
Dquantization_utils.h484 TypeAttr RescaleQuantizedType(Type input, Attribute factor);
498 TypeAttr GetQuantizedTypeAttr(Builder builder, Type input_type, Attribute min,
515 TypeAttr CastQuantizedTypeAttrFromExpressedType(Builder builder,
516 TypeAttr source, Type target,
/external/llvm-project/debuginfo-tests/llvm-prettyprinters/gdb/
Dmlir-support.cpp36 mlir::Attribute TypeAttr = mlir::TypeAttr::get(IndexType); variable
Dmlir-support.gdb98 # CHECK: cast<mlir::TypeAttr>
100 p TypeAttr
/external/llvm-project/mlir/lib/IR/
DFunctionSupport.cpp56 op->setAttr(getTypeAttrName(), TypeAttr::get(newType)); in eraseFunctionArguments()
92 op->setAttr(getTypeAttrName(), TypeAttr::get(newType)); in eraseFunctionResults()
DBuiltinDialect.cpp59 OpaqueElementsAttr, SparseElementsAttr, StringAttr, TypeAttr, in initialize()
99 state.addAttribute(getTypeAttrName(), TypeAttr::get(type)); in build()
/external/llvm-project/llvm/include/llvm/IR/
DAttributes.td29 class TypeAttr<string S> : Attr<S>;
52 def ByVal : TypeAttr<"byval">;
55 def ByRef : TypeAttr<"byref">;
161 def Preallocated : TypeAttr<"preallocated">;
207 def StructRet : TypeAttr<"sret">;
251 def MustProgress : TypeAttr<"mustprogress">;
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc149 cast_op.getInputElementType().cast<TypeAttr>().getValue(); in CastToNonDerivedType()
165 Type target_input_type = input_types[0].cast<TypeAttr>().getValue(); in CastValuesToSameType()
168 Type current_input_type = input_types[i].cast<TypeAttr>().getValue(); in CastValuesToSameType()
332 attrs.lookup(attr_key.getValue()).cast<TypeAttr>().getValue())); in DeriveOutputTypes()
345 attrs.lookup(attr_keys[1].getValue()).cast<TypeAttr>().getValue(); in DeriveOutputTypes()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc45 return mlir::TypeAttr::get(type); in ConvertNonFuncAttributeValue()
64 attrs.push_back(mlir::TypeAttr::get(type)); in ConvertNonFuncAttributeValue()
Dexport_utils.cc171 Status ConvertAttribute(const mlir::TypeAttr& type, bool remove_ref_type, in ConvertAttribute()
215 } else if (auto attr = a.dyn_cast<mlir::TypeAttr>()) { in ConvertAttribute()
385 .Case<mlir::ArrayAttr, mlir::TypeAttr>([&](auto derived_attr) { in ConvertAttributes()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_quantize_helper.h73 TypeAttr attr = op->template getAttrOfType<TypeAttr>( in GetIntermediateElementType()
438 TypeAttr attr = in processIntermediates()
439 op->template getAttrOfType<TypeAttr>(intermediate_attributes[index]); in processIntermediates()
488 TypeAttr::get(qtype.castFromExpressedType( in processIntermediates()
Ddefault_quant_params.cc180 TypeAttr type_attr = TypeAttr::get(new_type); in QuantizeValue()
/external/llvm-project/mlir/test/mlir-tblgen/
Dop-attribute.td113 TypeAttr:$type_attr
131 …F: if (!(((tblgen_some_type_attr.isa<::mlir::TypeAttr>())) && ((tblgen_some_type_attr.cast<::mlir:…
134 // DEF: if (!(((tblgen_type_attr.isa<::mlir::TypeAttr>())) && ((tblgen_type_attr.cast<::mlir::TypeA…
/external/llvm-project/mlir/include/mlir/Dialect/Tosa/Utils/
DQuantUtils.h63 TypeAttr buildQTypeAttrFromMinMax(OpBuilder builder, Type inputDType,
/external/llvm-project/clang/include/clang/AST/
DAttr.h114 class TypeAttr : public Attr {
116 TypeAttr(ASTContext &Context, const AttributeCommonInfo &CommonInfo, in TypeAttr() function
/external/llvm-project/mlir/lib/Dialect/Quant/IR/
DQuantOps.cpp46 if (auto typeAttr = quantSpec.dyn_cast<TypeAttr>()) { in isValidQuantizationSpec()
/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/lite/
Dtfl_to_std.cc58 TypeAttr::get(out_type)); in ConvertMlirQuantOpsToTFLQuantOps()
/external/clang/include/clang/Basic/
DAttr.td313 class TypeAttr : Attr {
373 def AddressSpace : TypeAttr {
718 def OpenCLPrivateAddressSpace : TypeAttr {
723 def OpenCLGlobalAddressSpace : TypeAttr {
728 def OpenCLLocalAddressSpace : TypeAttr {
733 def OpenCLConstantAddressSpace : TypeAttr {
738 def OpenCLGenericAddressSpace : TypeAttr {
977 def NeonPolyVectorType : TypeAttr {
983 def NeonVectorType : TypeAttr {
1113 def TypeNonNull : TypeAttr {
[all …]

1234