| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tpu_sharding_identification_pass.cc | 154 builder->getStringAttr(operand_sharding.getValue())); in IdentifyXlaShardingForComputationInputs()
164 builder->getStringAttr(kReplicateSharding)); in IdentifyXlaShardingForComputationInputs()
172 builder->getStringAttr(arg_sharding.getValue())); in IdentifyXlaShardingForComputationInputs()
179 builder->getStringAttr(logical_core_0_sharding)); in IdentifyXlaShardingForComputationInputs()
272 builder->getStringAttr(result_sharding.getValue())); in IdentifyXlaShardingForComputationOutputs()
281 builder->getStringAttr(kReplicateSharding)); in IdentifyXlaShardingForComputationOutputs()
288 builder->getStringAttr(retval_sharding.getValue())); in IdentifyXlaShardingForComputationOutputs()
295 builder->getStringAttr(logical_core_0_sharding)); in IdentifyXlaShardingForComputationOutputs()
|
| D | tf_device_assignment.cc | 42 op->setAttr("device", builder.getStringAttr(default_device_)); in runOnFunction()
47 op->setAttr("device", builder.getStringAttr(default_device_)); in runOnFunction()
|
| D | executor_tpuv1_outline_tpu_island.cc | 143 /*config=*/builder.getStringAttr(""), in runOnOperation()
144 /*config_proto=*/builder.getStringAttr(""), in runOnOperation()
145 /*executor_type=*/builder.getStringAttr("")); in runOnOperation()
|
| D | tpu_extract_outside_compilation.cc | 100 builder.getStringAttr(communication_key)); in CreateSendFromHostOp()
104 /*dynamic_key=*/compilation_key, builder.getStringAttr(communication_key), in CreateSendFromHostOp()
117 builder.getStringAttr(communication_key)); in CreateRecvAtHostOp()
121 builder.getStringAttr(communication_key), in CreateRecvAtHostOp()
187 loc_op->getLoc(), builder.getStringAttr(host_device), in CreateLaunchOpForOutsideCluster()
247 builder.getStringAttr(args_communication_key), in CreateHostCompute()
248 builder.getStringAttr(retvals_communication_key), in CreateHostCompute()
|
| D | readonly_references_to_resources.cc | 143 if (!device_attr) device_attr = builder.getStringAttr(""); in runOnFunction()
158 builder.getStringAttr(variable_name))}); in runOnFunction()
|
| D | cross_host_transfer.cc | 134 builder.getStringAttr(src_host + kCPUDevice)); in runOnFunction()
139 builder.getStringAttr(dst_host + kCPUDevice)); in runOnFunction()
|
| D | functional_control_flow_to_regions.cc | 101 builder.getStringAttr(if_op.then_function().getName()), in ConvertIfOp()
102 builder.getStringAttr(if_op.else_function().getName())); in ConvertIfOp()
|
| D | tpu_variable_runtime_reformatting.cc | 354 builder->getNamedAttr(kDeviceAttr, builder->getStringAttr(device)), in CreateStateVars()
355 builder->getNamedAttr("container", builder->getStringAttr("")), in CreateStateVars()
358 builder->getStringAttr(GetRandomStateVariableName()))})); in CreateStateVars()
369 loc, builder->getStringAttr(device), op->getResultTypes()); in WrapOpInLaunch()
|
| D | legalize_hlo.cc | 181 /*padding=*/rewriter.getStringAttr(padding), in CreateConvOp()
183 /*data_format=*/rewriter.getStringAttr("NHWC"), in CreateConvOp()
190 /*padding=*/rewriter.getStringAttr(padding), in CreateConvOp()
192 /*data_format=*/rewriter.getStringAttr("NHWC"), in CreateConvOp()
932 rewriter.getI64ArrayAttr(kstrides), rewriter.getStringAttr(padding), in replaceWithAvgPool()
933 rewriter.getStringAttr("NHWC")); in replaceWithAvgPool()
938 rewriter.getI64ArrayAttr(kstrides), rewriter.getStringAttr(padding), in replaceWithAvgPool()
939 rewriter.getStringAttr("NDHWC")); in replaceWithAvgPool()
|
| D | outside_compiled_to_host_launch.cc | 65 host_op->getLoc(), builder.getStringAttr(host_device), in WrapOpInLaunch()
|
| D | tpu_host_computation_expansion.cc | 111 builder->getStringAttr("")); in ExpandHeadOutsideCompiledOps()
|
| D | resource_device_inference.cc | 160 builder.getStringAttr(*device)); in ComputeResourceDevicesInComputation()
195 identity->setAttr(kDeviceAttr, builder.getStringAttr(*device)); in ComputeResourceDevicesInComputation()
|
| D | gpu_fusion.cc | 107 op->setAttr("activation_mode", rewriter.getStringAttr("Relu")); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_operator.cc | 54 return builder.getStringAttr(option_name); in GetPaddingAttr()
211 return builder.getStringAttr(option_name); in BuildTFL_AFAttr()
218 return builder.getStringAttr(option_name); in BuildTFL_FullyConnectedOptionsWeightFormatAttr()
224 return builder.getStringAttr(option_name); in BuildTFL_LSTMKernelTypeAttr()
230 return builder.getStringAttr(option_name); in BuildTFL_MirrorPaddingAttr()
236 return builder.getStringAttr(option_name); in BuildTFL_PaddingAttr()
244 builder.getNamedAttr("custom_code", builder.getStringAttr(custom_code))); in CustomOptionsToAttributes()
|
| /external/llvm-project/flang/lib/Lower/ |
| D | OpenMP.cpp | 196 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
201 firOpBuilder.getStringAttr(omp::stringifyClauseDefault( in genOMP()
205 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
209 parallelOp.default_valAttr(firOpBuilder.getStringAttr( in genOMP()
219 firOpBuilder.getStringAttr(omp::stringifyClauseProcBindKind( in genOMP()
224 firOpBuilder.getStringAttr(omp::stringifyClauseProcBindKind( in genOMP()
229 firOpBuilder.getStringAttr(omp::stringifyClauseProcBindKind( in genOMP()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 154 const StringAttr kLT = rewriter.getStringAttr( in MaterializeErfcApproximationF64ForMagnituteGEOne()
229 const StringAttr kLT = rewriter.getStringAttr( in MaterializeErfApproximationF64()
256 const StringAttr kLT = rewriter.getStringAttr( in MaterializeErfcApproximationF64()
292 const StringAttr kLT = rewriter.getStringAttr( in MaterializeErfcApproximationF32ForMagnitudeGEOne()
403 const StringAttr kLT = rewriter.getStringAttr( in MaterializeErfcApproximationF32()
521 const StringAttr kLT = rewriter.getStringAttr( in MaterializeLgamma()
666 const StringAttr kLT = rewriter.getStringAttr( in MaterializeDigamma()
749 const StringAttr kLE = rewriter.getStringAttr( in MaterializeDigamma()
752 const StringAttr kEQ = rewriter.getStringAttr( in MaterializeDigamma()
849 const StringAttr kLT = rewriter.getStringAttr( in MaterializeZeta()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_attr.cc | 37 return builder->getStringAttr(value.s()); in ConvertNonFuncAttributeValue()
56 attrs.push_back(builder->getStringAttr(item)); in ConvertNonFuncAttributeValue()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | optimize.cc | 371 auto none_af = rewriter.getStringAttr("NONE"); in matchAndRewrite()
381 auto none_af = rewriter.getStringAttr("NONE"); in matchAndRewrite()
394 rewriter.getStringAttr(add_op.fused_activation_function()), in matchAndRewrite()
395 /*weights_format=*/rewriter.getStringAttr(fc_op.weights_format()), in matchAndRewrite()
416 auto new_activation_func = rewriter.getStringAttr(Act); in matchAndRewrite()
418 rewriter.getStringAttr(fully_connected_op.weights_format()); in matchAndRewrite()
502 rewriter.getStringAttr(mul_op.fused_activation_function()), in matchAndRewrite()
503 /*weights_format=*/rewriter.getStringAttr(fc_op.weights_format()), in matchAndRewrite()
|
| D | legalize_tf.cc | 281 rewriter.getStringAttr("NONE"), rewriter.getStringAttr("DEFAULT"), in matchAndRewrite()
376 /*fused_activation_function=*/rewriter.getStringAttr("NONE"), padding, in matchAndRewrite()
499 rewriter.getStringAttr("TANH"))); in matchAndRewrite()
562 rewriter.getStringAttr("TANH"))); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/Conversion/ShapeToStandard/ |
| D | ShapeToStandard.td | 19 $_builder.getStringAttr("required broadcastable shapes")
|
| /external/llvm-project/mlir/test/lib/Transforms/ |
| D | TestLinalgTransforms.cpp | 310 procInfo[0] = {b.create<IdOp>(loc, indexType, b.getStringAttr("y")), in getGpuProcIds()
311 b.create<NProcsOp>(loc, indexType, b.getStringAttr("y"))}; in getGpuProcIds()
312 procInfo[1] = {b.create<IdOp>(loc, indexType, b.getStringAttr("x")), in getGpuProcIds()
313 b.create<NProcsOp>(loc, indexType, b.getStringAttr("x"))}; in getGpuProcIds()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ir/ |
| D | mlir_hlo_builder.cc | 126 builder_.getStringAttr(FftType_Name(fft_type)), in FftInternal()
149 loc_, ty, GetValues(operands), builder_.getStringAttr(call_target_name), in CustomCallInternal()
151 builder_.getStringAttr(opaque)); in CustomCallInternal()
417 builder_.getStringAttr(ComparisonDirectionToString(direction)), in Compare()
418 builder_.getStringAttr(ComparisonTypeToString(type))); in Compare()
452 builder_.getStringAttr( in TriangularSolveInternal()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | BuiltinDialect.cpp | 98 builder.getStringAttr(name)); in build()
212 mlir::SymbolTable::getSymbolAttrName(), builder.getStringAttr(*name))); in build()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | hlo_function_importer.cc | 322 builder_->getStringAttr(custom_call->custom_call_target()))); in ImportInstructionImpl()
328 builder_->getStringAttr(custom_call->raw_backend_config_string()))); in ImportInstructionImpl()
651 builder_->getStringAttr(TriangularSolveOptions::Transpose_Name( in ImportInstructionImpl()
727 builder_->getStringAttr(FftType_Name(instruction->fft_type())); in ImportInstructionImpl()
803 builder_->getStringAttr(xla::ToString(instruction->fusion_kind()))); in ImportInstructionImpl()
901 builder_->getStringAttr(ComparisonDirectionToString(direction))); in ConvertComparisonDirection()
907 "compare_type", builder_->getStringAttr(ComparisonTypeToString(type))); in ConvertComparisonType()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/IR/ |
| D | chlo_canonicalize.td | 23 "$_builder.getStringAttr(\",\" + $0.getValue().str())">;
|