/external/llvm-project/mlir/test/lib/Transforms/ |
D | TestGpuMemoryPromotion.cpp | 39 if (op.getArgAttrOfType<UnitAttr>(i, "gpu.test_promote_workgroup")) in runOnOperation()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
D | deduplicate_bound_input_bindings.cc | 41 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in runOnFunction()
|
D | tensor_device_copy_conversion.cc | 53 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
|
D | cross_host_transfer.cc | 101 if (StringAttr device_attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
|
D | optimize_global_tensors.cc | 209 func.getArgAttrOfType<SymbolRefAttr>(i, "tf_saved_model.bound_input"); in CreateGlobalTensorUsesMap()
|
D | lift_variables.cc | 160 func.getArgAttrOfType<StringAttr>(i, kResourceNameArgAttr); in LiftVariables()
|
D | tpu_merge_variables_with_execute.cc | 167 auto resource_attr = func.getArgAttrOfType<mlir::StringAttr>( in BuildVariableAccessInfo() 464 auto aliasing_attr = main_func.getArgAttrOfType<mlir::IntegerAttr>( in UpdateSerializedModule()
|
D | tpu_device_propagation.cc | 144 func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); in PropagateDevicesFromArguments()
|
D | resource_device_inference.cc | 131 func.getArgAttrOfType<mlir::StringAttr>(arg_no, kFuncDeviceAttr); in GetDeviceAttr()
|
D | tpu_dynamic_layout_pass.cc | 120 device = func.getArgAttrOfType<mlir::StringAttr>(arg.getArgNumber(), in IsSupportedInputOp()
|
D | cluster_tf_ops_pass.cc | 126 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in GetFunctionMetadatas()
|
/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 | 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/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
D | raise_to_tf.cc | 233 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()
|
D | decompose.cc | 173 auto attr_name = compose_func.getArgAttrOfType<StringAttr>( in RewriteUnregisteredTFOps()
|
/external/llvm-project/mlir/lib/Conversion/GPUToSPIRV/ |
D | ConvertGPUToSPIRV.cpp | 225 if (funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in getDefaultABIAttrs() 249 auto abiAttr = funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/ |
D | tf_saved_model.cc | 355 if (auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in VerifyExportedFunc() 439 auto attr = func.getArgAttrOfType<FlatSymbolRefAttr>( in LookupBoundInput()
|
/external/llvm-project/mlir/include/mlir/IR/ |
D | FunctionSupport.h | 321 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/ |
D | LowerABIAttributesPass.cpp | 182 auto abiInfo = funcOp.getArgAttrOfType<spirv::InterfaceVarABIAttr>( in matchAndRewrite()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
D | compile_mlir_util.cc | 110 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/ |
D | export_graphdef.cc | 239 func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) in GetArgumentNode() 680 function.getArgAttrOfType<mlir::IntegerAttr>( in ConvertLibFunction()
|
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/analysis/ |
D | resource_alias_analysis.cc | 277 auto id_attr = func_op.getArgAttrOfType<IntegerAttr>( in ResourceAliasAnalysisInfo()
|
/external/llvm-project/mlir/lib/Target/LLVMIR/ |
D | ModuleTranslation.cpp | 912 if (auto attr = func.getArgAttrOfType<BoolAttr>( in convertOneFunction() 924 if (auto attr = func.getArgAttrOfType<IntegerAttr>( in convertOneFunction()
|
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/ |
D | tfr_ops.cc | 217 func.getArgAttrOfType<StringAttr>(arg.index(), kAttrArgumentNameAttr); in Verify()
|