Searched refs:setResultAttr (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_sharding_identification_pass.cc | 271 func.setResultAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationOutputs()
280 func.setResultAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationOutputs()
287 func.setResultAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationOutputs()
294 func.setResultAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationOutputs()
|
| D | tpu_device_propagation.cc | 221 func.setResultAttr(operand.getOperandNumber(), kFuncDeviceAttr, in PropagateDevicesToResults()
|
| D | promote_resources_to_args.cc | 349 function.setResultAttr(resource.first, kResourceNameArgAttr, in PromoteResourcesToArguments()
|
| D | cluster_tf_ops_pass.cc | 216 func_op.setResultAttr( in CreateFunctions()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | FunctionSupport.h | 403 void setResultAttr(unsigned index, Identifier name, Attribute value);
404 void setResultAttr(unsigned index, StringRef name, Attribute value) { in setResultAttr() function
405 setResultAttr(index, in setResultAttr()
651 void FunctionLike<ConcreteType>::setResultAttr(unsigned index, Identifier name, in setResultAttr() function
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1507 func.setResultAttr(index, dialect_attribute, converted_attr); in ConvertFunctionArgAndRets()
3193 func.setResultAttr(index_path.index(), "tf_saved_model.index_path", in CreateSavedModelIR()
3679 func_op.setResultAttr( in ConvertSignature()
|