| /external/tensorflow/tensorflow/compiler/mlir/tfr/utils/ |
| D | utils.cc | 42 void CollectAllowedAttrs(CallOp src, NamedAttrList* attrs) { in CollectAllowedAttrs()
53 const NamedAttrList& attrs) { in AddAttributesInSameBlock()
64 const NamedAttrList& attrs) { in AddAttributes()
167 NamedAttrList attrs; in CopyNonSymbolRefAttrs()
181 NamedAttrList attrs; in PropagateAttrsToOperations()
|
| /external/tensorflow/tensorflow/core/ir/importexport/ |
| D | functiondef_import.cc | 220 tensorflow::StatusOr<NamedAttrList> ConvertArgDefAttributes( in ConvertArgDefAttributes()
222 NamedAttrList input_attrs; in ConvertArgDefAttributes()
269 NamedAttrList attrs; in ImportGenericFunction()
286 NamedAttrList attr_defs; in ImportGenericFunction()
288 NamedAttrList attr_def; in ImportGenericFunction()
360 TF_ASSIGN_OR_RETURN(NamedAttrList input_attrs, in ImportGenericFunction()
364 NamedAttrList arg_attr; in ImportGenericFunction()
376 args_attrs.push_back(NamedAttrList{}.getDictionary(context)); in ImportGenericFunction()
386 TF_ASSIGN_OR_RETURN(NamedAttrList output_attrs, in ImportGenericFunction()
393 NamedAttrList output_attrs; in ImportGenericFunction()
|
| D | graphdef_import.cc | 203 StatusOr<unsigned> ArgNumType(const NamedAttrList &attrs,
209 const FunctionDef &function, GraphFuncOp op, NamedAttrList &attrs);
212 NamedAttrList &attrs);
356 const FunctionDef &function, GraphFuncOp op, NamedAttrList &attrs) { in ConvertFunctionAttributes()
412 NamedAttrList &attrs) { in ConvertArgumentAttributes()
546 NamedAttrList func_attrs; in ConvertFunctionDef()
562 NamedAttrList attrs; in ConvertFunctionDef()
595 NamedAttrList attrs; in ConvertFunctionDef()
695 StatusOr<unsigned> GraphDefImporter::ArgNumType(const NamedAttrList &attrs, in ArgNumType()
|
| D | convert_attributes.cc | 351 NamedAttrList subattrs; in ConvertNonFuncAttributeValue()
379 NamedAttrList attrs; in ConvertAttributeValue()
|
| /external/tensorflow/tensorflow/core/transforms/utils/ |
| D | utils.h | 26 class NamedAttrList; variable
35 void EraseRegularNodeAttributes(NamedAttrList &attr_list);
|
| D | utils.cc | 45 void EraseRegularNodeAttributes(NamedAttrList &attr_list) { in EraseRegularNodeAttributes()
46 NamedAttrList new_attr_list; in EraseRegularNodeAttributes()
54 NamedAttrList dst_attrs(dst->getAttrDictionary()); in ForwardNonIntrinsicAttributes()
|
| /external/tensorflow/tensorflow/core/transforms/graph_compactor/ |
| D | pass.cc | 103 NamedAttrList attrs = func->getAttrDictionary(); in runOnOperation()
110 NamedAttrList attrs = func.arg_attrsAttr()[i].cast<DictionaryAttr>(); in runOnOperation()
120 for (NamedAttrList attrs : in runOnOperation()
137 for (NamedAttrList attrs : control_attrs.getAsRange<DictionaryAttr>()) { in runOnOperation()
308 NamedAttrList attrs = op->getAttrDictionary(); in addDefaultValuedAttrs()
|
| /external/tensorflow/tensorflow/core/transforms/consolidate_attrs/ |
| D | pass.cc | 123 Type refineTypeWithOutputShapes(Type type, NamedAttrList &attrs);
130 Type type, NamedAttrList &attrs) { in refineTypeWithOutputShapes()
179 NamedAttrList attrs(func.getArgAttrs(arg.getArgNumber())); in reifyAndDropFunctionArgumentAttributes()
202 NamedAttrList attrs(it.value()); in reifyAndDropFunctionResultAttributes()
374 NamedAttrList attrs(func->getAttrDictionary()); in runOnOperation()
403 NamedAttrList attrs(func->getAttrDictionary()); in prepareFunctionAttributes()
440 NamedAttrList attrs(attr_dict); in prepareAttributesFor()
507 NamedAttrList attrs(op->getAttrDictionary()); in matchAndRewrite()
529 NamedAttrList attrs(op->getAttrDictionary()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfrt/ir/ |
| D | tfrt_fallback_common.cc | 51 mlir::NamedAttrList op_attrs; in ParseExecuteOpCommon()
52 mlir::NamedAttrList op_func_attrs; in ParseExecuteOpCommon()
88 mlir::NamedAttrList attrs; in ParseExecuteOpCommon()
|
| D | tfrt_fallback_async.cc | 217 NamedAttrList op_attrs; in parse()
233 mlir::NamedAttrList attrs; in parse()
|
| /external/tensorflow/tensorflow/core/runtime_fallback/opdefs/ |
| D | tfrt_fallback_common.cc | 51 mlir::NamedAttrList op_attrs; in ParseExecuteOpCommon()
52 mlir::NamedAttrList op_func_attrs; in ParseExecuteOpCommon()
88 mlir::NamedAttrList attrs; in ParseExecuteOpCommon()
|
| D | tfrt_fallback_async.cc | 151 NamedAttrList op_attrs; in parseBatchFunctionOp()
167 mlir::NamedAttrList attrs; in parseBatchFunctionOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 105 SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs,
118 const SmallVectorImpl<Value>& inputs, const NamedAttrList& attr_list,
230 SmallVectorImpl<Value>* inputs, NamedAttrList* arg_attrs, in CollectInputsAndAttributes()
377 const SmallVectorImpl<Value>& inputs, const NamedAttrList& attr_list, in CreateAndReplaceOp()
435 NamedAttrList argument_attrs; in matchAndRewrite()
|
| /external/tensorflow/tensorflow/core/transforms/region_to_functional/ |
| D | impl.cc | 139 NamedAttrList BuildAttributes(RegionAttr preserved, ValueRange arguments,
156 TypeRange res_types, NamedAttrList attrs) const;
500 NamedAttrList BasePattern::BuildAttributes(RegionAttr preserved, in BuildAttributes()
504 NamedAttrList attrs(preserved ? preserved.getAttrs() : DictionaryAttr()); in BuildAttributes()
518 NamedAttrList attrs(attr ? attr[it.index()].template cast<DictionaryAttr>() in BuildAttributes()
586 NamedAttrList ctl_attrs; in GetControlRetAttrs()
598 NamedAttrList attrs) const { in CreateFunc()
645 NamedAttrList func_attrs = BuildAttributes( in Outline()
960 NamedAttrList cond_attrs = this->BuildAttributes( in matchAndRewrite()
|
| /external/tensorflow/tensorflow/dtensor/mlir/ |
| D | shape_utils.cc | 111 mlir::NamedAttrList GetAllAttributesFromOperation(mlir::Operation* op) { in GetAllAttributesFromOperation()
112 mlir::NamedAttrList attr_list; in GetAllAttributesFromOperation()
|
| D | handle_sparsetensors.cc | 123 mlir::NamedAttrList attributes(dict_attr); in UpdateFunctionInputAttributes()
|
| /external/tensorflow/tensorflow/core/ir/utils/ |
| D | shape_inference_utils.cc | 71 NamedAttrList GetAllAttributesFromOperation(Operation* op) { in GetAllAttributesFromOperation()
72 NamedAttrList attr_list; in GetAllAttributesFromOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_cluster_formation.cc | 70 using MetadataMap = llvm::SmallDenseMap<llvm::StringRef, NamedAttrList, 8>;
98 NamedAttrList attrs(metadata_op->getAttrDictionary()); in CollectMetadata()
702 NamedAttrList attributes; in runOnOperation()
|
| D | tfg-to-tfe.cc | 65 mlir::NamedAttrList list; in FilterTfgSpecificArgResultAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 101 mlir::NamedAttrList attrs; in ConvertAttributeValue()
|
| /external/tensorflow/tensorflow/security/advisory/ |
| D | tfsa-2022-140.md | 10 StatusOr<unsigned> GraphDefImporter::ArgNumType(const NamedAttrList &attrs,
|
| /external/tensorflow/tensorflow/core/ir/ |
| D | ops.cc | 567 NamedAttrList attrs; in parse()
600 NamedAttrList result_attr; in parse()
615 NamedAttrList parsed_attributes; in parse()
804 NamedAttrList attrs; in parse()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 143 using mlir::NamedAttrList;
362 NamedAttrList* attributes);
1304 NamedAttrList* attributes) { in ConvertFunctionCallAttribute()
1338 NamedAttrList attrs; in ConvertAttributeValue()
1609 llvm::SmallVector<mlir::NamedAttrList, 4> arg_attrs; in ConvertFunctionArgAndRets()
1611 llvm::SmallVector<mlir::NamedAttrList, 4> ret_attrs; in ConvertFunctionArgAndRets()
1721 llvm::to_vector<4>(llvm::map_range(arg_attrs, [&](NamedAttrList& list) { in ConvertFunctionArgAndRets()
1725 llvm::to_vector<4>(llvm::map_range(ret_attrs, [&](NamedAttrList& list) { in ConvertFunctionArgAndRets()
|
| /external/tensorflow/tensorflow/core/transforms/functional_to_region/ |
| D | impl.cc | 187 NamedAttrList preserved_attrs; in PreserveFunctionAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/transforms/ |
| D | mhlo_to_lhlo_with_xla.cc | 936 NamedAttrList attrs(op->getAttrDictionary()); in EmitDnnConvolution()
1631 NamedAttrList arg_attr_list; in Initialize()
|