| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_saved_model.cc | 86 sym_ref.cast<FlatSymbolRefAttr>().getValue()); in Verify()
191 if (!named_attr.second.isa<FlatSymbolRefAttr>()) { in verifyRegionArgAttribute()
195 auto symbol_name = named_attr.second.cast<FlatSymbolRefAttr>().getValue(); in verifyRegionArgAttribute()
299 return sym_ref.cast<FlatSymbolRefAttr>().getValue() == func.getName(); in VerifySavedModelModule()
311 symbol_use.getSymbolRef().cast<FlatSymbolRefAttr>().getValue()); in VerifySavedModelModule()
355 if (auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in VerifyExportedFunc()
439 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in LookupBoundInput()
464 sym_ref.cast<FlatSymbolRefAttr>().getValue()); in matchAndRewrite()
507 sym_ref.cast<FlatSymbolRefAttr>().getValue()); in GetSessionInitializerExportedName()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | executor_tpuv1_outline_tpu_island.cc | 53 void MoveFuncOp(FlatSymbolRefAttr &symbol_ref, SymbolTable &from, in MoveFuncOp()
155 if (auto symbol_ref = attr.second.dyn_cast<FlatSymbolRefAttr>()) { in runOnOperation()
161 auto symbol_ref = attribute.dyn_cast<FlatSymbolRefAttr>(); in runOnOperation()
|
| D | resource_op_lifting_cleanup.cc | 145 auto symref = attr.second.dyn_cast<FlatSymbolRefAttr>(); in EliminateUnusedResultsForIfCase()
149 FlatSymbolRefAttr::get(op->getContext(), cloned.getName())); in EliminateUnusedResultsForIfCase()
220 op.condAttr(FlatSymbolRefAttr::get(op.getContext(), cloned_cond.getName())); in EliminateUnusedResultsForWhile()
221 op.bodyAttr(FlatSymbolRefAttr::get(op.getContext(), cloned_body.getName())); in EliminateUnusedResultsForWhile()
|
| D | deduplicate_bound_input_bindings.cc | 41 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in runOnFunction()
|
| D | remove_vars_in_session_initializer.cc | 88 sym_ref.cast<FlatSymbolRefAttr>().getValue()); in runOnOperation()
|
| D | optimize_global_tensors.cc | 214 sym.cast<FlatSymbolRefAttr>().getValue()); in CreateGlobalTensorUsesMap()
|
| D | lift_variables.cc | 165 FlatSymbolRefAttr::get(context, resource_name); in LiftVariables()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | SymbolTable.cpp | 49 auto leafRef = FlatSymbolRefAttr::get(symbolName, ctx); in collectValidReferencesFor()
58 SmallVector<FlatSymbolRefAttr, 1> nestedRefs(1, leafRef); in collectValidReferencesFor()
76 FlatSymbolRefAttr::get(*symbolTableName, ctx)); in collectValidReferencesFor()
314 ArrayRef<FlatSymbolRefAttr> nestedRefs = symbol.getNestedReferences(); in lookupSymbolInImpl()
324 for (FlatSymbolRefAttr ref : nestedRefs.drop_back()) { in lookupSymbolInImpl()
661 if (ref.isa<FlatSymbolRefAttr>() || in isReferencePrefixOf()
825 FlatSymbolRefAttr newLeafAttr) { in generateNewRefAttr()
826 if (oldAttr.isa<FlatSymbolRefAttr>()) in generateNewRefAttr()
858 FlatSymbolRefAttr newLeafAttr = FlatSymbolRefAttr::get(newSymbol, ctx); in replaceAllSymbolUsesImpl()
872 if (scope.symbol.isa<FlatSymbolRefAttr>()) { in replaceAllSymbolUsesImpl()
|
| D | AttributeDetail.h | 295 FlatSymbolRefAttr> {
296 using KeyTy = std::pair<StringRef, ArrayRef<FlatSymbolRefAttr>>;
309 auto size = SymbolRefAttributeStorage::totalSizeToAlloc<FlatSymbolRefAttr>( in construct()
315 result->getTrailingObjects<FlatSymbolRefAttr>()); in construct()
320 ArrayRef<FlatSymbolRefAttr> getNestedRefs() const { in getNestedRefs()
321 return {getTrailingObjects<FlatSymbolRefAttr>(), numNestedRefs}; in getNestedRefs()
|
| D | Builders.cpp | 206 FlatSymbolRefAttr Builder::getSymbolRefAttr(Operation *value) { in getSymbolRefAttr()
212 FlatSymbolRefAttr Builder::getSymbolRefAttr(StringRef value) { in getSymbolRefAttr()
217 ArrayRef<FlatSymbolRefAttr> nestedReferences) { in getSymbolRefAttr()
|
| /external/llvm-project/mlir/lib/CAPI/IR/ |
| D | BuiltinAttributes.cpp | 210 SmallVector<FlatSymbolRefAttr, 4> refs; in mlirSymbolRefAttrGet()
213 refs.push_back(unwrap(references[i]).cast<FlatSymbolRefAttr>()); in mlirSymbolRefAttrGet()
240 return unwrap(attr).isa<FlatSymbolRefAttr>(); in mlirAttributeIsAFlatSymbolRef()
244 return wrap(FlatSymbolRefAttr::get(unwrap(symbol), unwrap(ctx))); in mlirFlatSymbolRefAttrGet()
248 return wrap(unwrap(attr).cast<FlatSymbolRefAttr>().getValue()); in mlirFlatSymbolRefAttrGetValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tf_framework_legalize_to_llvm.cc | 47 FlatSymbolRefAttr getOrInsertTFFunction(PatternRewriter &rewriter, in getOrInsertTFFunction()
106 FlatSymbolRefAttr tf_func_ref = getOrInsertTFFunction(rewriter, op); in matchAndRewrite()
230 FlatSymbolRefAttr tf_func_ref = getOrInsertTFFunction(rewriter, op); in matchAndRewrite()
262 FlatSymbolRefAttr tf_func_ref = getOrInsertTFFunction(rewriter, op); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | export_utils.cc | 124 Status ConvertAttribute(const mlir::FlatSymbolRefAttr& attr, AttrValue* value) { in ConvertAttribute()
132 ConvertAttribute(attr.GetName().cast<mlir::FlatSymbolRefAttr>(), value)); in ConvertAttribute()
211 } else if (auto attr = a.dyn_cast<mlir::FlatSymbolRefAttr>()) { in ConvertAttribute()
364 ConvertAttribute(symbol_ref.cast<mlir::FlatSymbolRefAttr>(), &value)); in ConvertAttributes()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | remove_unused_func_args_tensors.cc | 51 if (auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in runOnOperation()
|
| D | insert_call_once_op.cc | 45 sym_ref.cast<FlatSymbolRefAttr>().getValue()); in runOnOperation()
|
| D | legalize_variables.cc | 62 auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in GetGlobalTensor()
|
| D | initialize_variables.cc | 56 auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in GetGlobalTensor()
|
| /external/llvm-project/mlir/lib/Conversion/LinalgToStandard/ |
| D | LinalgToStandard.cpp | 48 static FlatSymbolRefAttr getLibraryCallSymbolRef(Operation *op, in getLibraryCallSymbolRef()
58 FlatSymbolRefAttr fnNameAttr = rewriter.getSymbolRefAttr(fnName); in getLibraryCallSymbolRef()
|
| /external/llvm-project/mlir/test/lib/IR/ |
| D | TestFunc.cpp | 71 auto sym = func->getAttrOfType<FlatSymbolRefAttr>("test.set_type_from"); in runOnOperation()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 98 FlatSymbolRefAttr getSymbolRefAttr(Operation *value);
99 FlatSymbolRefAttr getSymbolRefAttr(StringRef value);
101 ArrayRef<FlatSymbolRefAttr> nestedReferences);
|
| D | BuiltinAttributes.h | 340 class FlatSymbolRefAttr; variable
351 static FlatSymbolRefAttr get(StringRef value, MLIRContext *ctx);
356 ArrayRef<FlatSymbolRefAttr> references,
369 ArrayRef<FlatSymbolRefAttr> getNestedReferences() const;
374 class FlatSymbolRefAttr : public SymbolRefAttr {
380 static FlatSymbolRefAttr get(StringRef value, MLIRContext *ctx) { in get()
|
| /external/llvm-project/mlir/test/mlir-tblgen/ |
| D | op-side-effects.td | 17 Arg<FlatSymbolRefAttr, "", [MemWrite]>:$flat_symbol,
|
| /external/llvm-project/mlir/lib/Target/LLVMIR/ |
| D | ConvertFromLLVMIR.cpp | 66 FlatSymbolRefAttr getPersonalityAsAttr(llvm::Function *f);
344 if (auto symbolRef = attr.dyn_cast<FlatSymbolRefAttr>()) in processConstant()
753 FlatSymbolRefAttr Importer::getPersonalityAsAttr(llvm::Function *f) { in getPersonalityAsAttr()
772 return FlatSymbolRefAttr(); in getPersonalityAsAttr()
789 if (FlatSymbolRefAttr personality = getPersonalityAsAttr(f)) in processFunction()
|
| /external/llvm-project/mlir/examples/toy/Ch7/mlir/ |
| D | LowerToLLVM.cpp | 106 static FlatSymbolRefAttr getOrInsertPrintf(PatternRewriter &rewriter, in getOrInsertPrintf()
|
| /external/llvm-project/mlir/examples/toy/Ch6/mlir/ |
| D | LowerToLLVM.cpp | 106 static FlatSymbolRefAttr getOrInsertPrintf(PatternRewriter &rewriter, in getOrInsertPrintf()
|