Home
last modified time | relevance | path

Searched refs:getArgAttrOfType (Results 1 – 25 of 28) sorted by relevance

12

/external/llvm-project/mlir/test/lib/Transforms/
DTestGpuMemoryPromotion.cpp39 if (op.getArgAttrOfType<UnitAttr>(i, "gpu.test_promote_workgroup")) in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Ddeduplicate_bound_input_bindings.cc41 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in runOnFunction()
Dtensor_device_copy_conversion.cc53 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
Dcross_host_transfer.cc101 if (StringAttr device_attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
Doptimize_global_tensors.cc209 func.getArgAttrOfType<SymbolRefAttr>(i, "tf_saved_model.bound_input"); in CreateGlobalTensorUsesMap()
Dlift_variables.cc160 func.getArgAttrOfType<StringAttr>(i, kResourceNameArgAttr); in LiftVariables()
Dtpu_merge_variables_with_execute.cc167 auto resource_attr = func.getArgAttrOfType<mlir::StringAttr>( in BuildVariableAccessInfo()
464 auto aliasing_attr = main_func.getArgAttrOfType<mlir::IntegerAttr>( in UpdateSerializedModule()
Dtpu_device_propagation.cc144 func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); in PropagateDevicesFromArguments()
Dresource_device_inference.cc131 func.getArgAttrOfType<mlir::StringAttr>(arg_no, kFuncDeviceAttr); in GetDeviceAttr()
Dtpu_dynamic_layout_pass.cc120 device = func.getArgAttrOfType<mlir::StringAttr>(arg.getArgNumber(), in IsSupportedInputOp()
Dcluster_tf_ops_pass.cc126 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in GetFunctionMetadatas()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dremove_unused_func_args_tensors.cc51 if (auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in runOnOperation()
Dlegalize_variables.cc62 auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in GetGlobalTensor()
Dinitialize_variables.cc56 auto sym = func.template getArgAttrOfType<FlatSymbolRefAttr>( in GetGlobalTensor()
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc233 auto attr_name = signature.getArgAttrOfType<StringAttr>( in CollectInputsAndAttributes()
296 auto attr_name = signature.getArgAttrOfType<StringAttr>( in CollectInputsAndAttributes()
298 auto attr_type = signature.getArgAttrOfType<StringAttr>( in CollectInputsAndAttributes()
Ddecompose.cc173 auto attr_name = compose_func.getArgAttrOfType<StringAttr>( in RewriteUnregisteredTFOps()
/external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/
DConvertGPUToSPIRV.cpp225 if (funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in getDefaultABIAttrs()
249 auto abiAttr = funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc355 if (auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in VerifyExportedFunc()
439 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in LookupBoundInput()
/external/llvm-project/mlir/include/mlir/IR/
DFunctionSupport.h321 AttrClass getArgAttrOfType(unsigned index, Identifier name) { in getArgAttrOfType() function
325 AttrClass getArgAttrOfType(unsigned index, StringRef name) { in getArgAttrOfType() function
/external/llvm-project/mlir/lib/Dialect/SPIRV/Transforms/
DLowerABIAttributesPass.cpp182 auto abiInfo = funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dcompile_mlir_util.cc110 main_func.getArgAttrOfType<mlir::StringAttr>(i, "mhlo.sharding"); in GetXlaInputShapes()
158 if (auto aliasing_output = main_func.getArgAttrOfType<mlir::IntegerAttr>( in GetOutputInfo()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dexport_graphdef.cc239 func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) in GetArgumentNode()
680 function.getArgAttrOfType<mlir::IntegerAttr>( in ConvertLibFunction()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/
Dresource_alias_analysis.cc277 auto id_attr = func_op.getArgAttrOfType<IntegerAttr>( in ResourceAliasAnalysisInfo()
/external/llvm-project/mlir/lib/Target/LLVMIR/
DModuleTranslation.cpp912 if (auto attr = func.getArgAttrOfType<BoolAttr>( in convertOneFunction()
924 if (auto attr = func.getArgAttrOfType<IntegerAttr>( in convertOneFunction()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc217 func.getArgAttrOfType<StringAttr>(arg.index(), kAttrArgumentNameAttr); in Verify()

12