Searched refs:setArgAttr (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_sharding_identification_pass.cc | 153 func.setArgAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationInputs()
163 func.setArgAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationInputs()
171 func.setArgAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationInputs()
178 func.setArgAttr(index, kShardingAttr, in IdentifyXlaShardingForComputationInputs()
|
| D | promote_resources_to_args.cc | 321 function.setArgAttr( in PromoteResourcesToArguments()
355 function.setArgAttr(input_output.first, "tf.aliasing_output", in PromoteResourcesToArguments()
406 function.setArgAttr(var_name_and_index.index() + var_handle_args_offset, in runOnOperation()
|
| D | annotate_parameter_replication.cc | 86 func.setArgAttr(entry.index(), kReplicationAttr, builder.getUnitAttr()); in runOnOperation()
|
| D | tpu_dynamic_padding_mapper.cc | 175 func.setArgAttr( in AnnotateFunctionArgumentsWithPaddings()
|
| D | lift_variables.cc | 168 func.setArgAttr(i, kSavedModelArgAttr, flat_symbol_ref_attr); in LiftVariables()
|
| D | resource_device_inference.cc | 159 func_op.setArgAttr(arg.getArgNumber(), kFuncDeviceAttr, in ComputeResourceDevicesInComputation()
|
| D | cluster_tf_ops_pass.cc | 212 func_op.setArgAttr(i, kTFDeviceAttr, in CreateFunctions()
|
| D | tpu_merge_variables_with_execute.cc | 469 main_func.setArgAttr( in UpdateSerializedModule()
|
| /external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/ |
| D | tensorflow_abi_knowledge_propagation.cc | 109 kernel.setArgAttr( in runOnFunction()
142 kernel.setArgAttr(kernel_p + 1, in runOnFunction()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | FunctionSupport.h | 340 void setArgAttr(unsigned index, Identifier name, Attribute value);
341 void setArgAttr(unsigned index, StringRef name, Attribute value) { in setArgAttr() function
342 setArgAttr(index, Identifier::get(name, this->getOperation()->getContext()), in setArgAttr()
591 void FunctionLike<ConcreteType>::setArgAttr(unsigned index, Identifier name, in setArgAttr() function
|
| /external/llvm-project/mlir/lib/Dialect/SPIRV/ |
| D | SPIRVLowering.cpp | 657 funcOp.setArgAttr(argIndex, argABIAttrName, argABIInfo[argIndex]); in setABIAttrs()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 3167 func.setArgAttr(index_path.index(), "tf_saved_model.index_path", in CreateSavedModelIR()
3176 func.setArgAttr(arg_index, "tf_saved_model.bound_input", symbol_ref); in CreateSavedModelIR()
3603 init_func_op.setArgAttr(iter.index(), "tf_saved_model.bound_input", in ConvertInitializer()
3675 func_op.setArgAttr(input_and_idx.index(), "tf_saved_model.index_path", in ConvertSignature()
|