Searched refs:getArgAttr (Results 1 – 10 of 10) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
| D | tf_saved_model.cc | 227 if (func.getArgAttr(i, "tf_saved_model.index_path") || in HasAnyTfSavedModelArgAttr()
228 func.getArgAttr(i, "tf_saved_model.bound_input")) { in HasAnyTfSavedModelArgAttr()
332 if (func.getArgAttr(i, "tf_saved_model.bound_input")) { in VerifyExportedFunc()
336 if (func.getArgAttr(i, "tf_saved_model.index_path")) { in VerifyExportedFunc()
344 if (func.getArgAttr(i, "tf.resource_name")) { in VerifyExportedFunc()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | FunctionSupport.h | 311 Attribute getArgAttr(unsigned index, Identifier name) { in getArgAttr() function
315 Attribute getArgAttr(unsigned index, StringRef name) { in getArgAttr() function
322 return getArgAttr(index, name).template dyn_cast_or_null<AttrClass>(); in getArgAttrOfType()
326 return getArgAttr(index, name).template dyn_cast_or_null<AttrClass>(); in getArgAttrOfType()
|
| /external/llvm-project/mlir/test/lib/IR/ |
| D | TestFunc.cpp | 24 if (func.getArgAttr(argIndex, "test.erase_this_arg")) { in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | gpu_compiler.cc | 1014 func.getArgAttr(i, "lmhlo.alloc").dyn_cast_or_null<mlir::IntegerAttr>(); in GetMlirAllocationInfo()
1047 func.getArgAttr(i, "lmhlo.alloc").cast<mlir::IntegerAttr>().getInt(); in GetMlirAllocationInfo()
1048 if (auto param_attr = func.getArgAttr(i, "lmhlo.params")) { in GetMlirAllocationInfo()
1051 static_cast<bool>(func.getArgAttr(i, "lmhlo.output_index"))); in GetMlirAllocationInfo()
1053 if (auto output_index_attr = func.getArgAttr(i, "lmhlo.output_index")) { in GetMlirAllocationInfo()
1071 if (auto param_attr = func.getArgAttr(i, "lmhlo.params")) { in GetMlirAllocationInfo()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | optimize_global_tensors.cc | 260 if (func.getArgAttr(i, "tf_saved_model.bound_input") && in EraseUnusedBoundInputs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | decompose.cc | 178 compose_func.getArgAttr(arg.index(), kAttrArgumentDefaultAttr); in RewriteUnregisteredTFOps()
|
| D | raise_to_tf.cc | 236 signature.getArgAttr(operand.index(), kAttrArgumentDefaultAttr); in CollectInputsAndAttributes()
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/ |
| D | LowerABIAttributesPass.cpp | 256 if (op.getArgAttr(i, attrName)) in runOnOperation()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
| D | resource_alias_analysis.cc | 270 return func_op.getArgAttr(arg.getArgNumber(), kResourceArgUniqueIdAttr); in ResourceAliasAnalysisInfo()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | mlir_hlo_to_hlo.cc | 1672 auto padding_map_attr = entry_func.getArgAttr(i, kPaddingMapAttr); in AddDynamicParameterBindings()
|