Home
last modified time | relevance | path

Searched refs:StringAttr (Results 1 – 25 of 165) sorted by relevance

1234567

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtensor_device_copy_conversion.cc43 StringAttr empty_string = StringAttr::get(func_op.getContext(), ""); in runOnFunction()
45 StringAttr arg_device = empty_string; in runOnFunction()
53 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
57 } else if (StringAttr attr = in runOnFunction()
58 arg.getDefiningOp()->getAttrOfType<StringAttr>( in runOnFunction()
63 StringAttr op_device = op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnFunction()
Dexecutor_tpuv1_island_coarsening.cc120 LogicalResult MergeIsland(llvm::function_ref<bool(StringAttr, Operation*)> in MergeIsland() argument
130 StringAttr cluster_name = in MergeIsland()
131 wrapped_op.getAttrOfType<StringAttr>(kTpuReplicateAttr); in MergeIsland()
133 cluster_name = wrapped_op.getAttrOfType<StringAttr>(kTpuStatusAttr); in MergeIsland()
150 StringAttr candidate_cluster_name = in MergeIsland()
151 candidate_wrapped_op.getAttrOfType<StringAttr>(kTpuReplicateAttr); in MergeIsland()
154 candidate_wrapped_op.getAttrOfType<StringAttr>(kTpuStatusAttr); in MergeIsland()
270 StringAttr cluster_name = in runOnOperation()
271 op->getAttrOfType<StringAttr>(kTpuReplicateAttr); in runOnOperation()
273 cluster_name = op->getAttrOfType<StringAttr>(kTpuStatusAttr); in runOnOperation()
[all …]
Dcross_host_transfer.cc88 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction()
101 if (StringAttr device_attr = func_op.getArgAttrOfType<StringAttr>( in runOnFunction()
107 if (StringAttr device_attr = in runOnFunction()
108 defining_op->getAttrOfType<StringAttr>(kOpDeviceAttr)) { in runOnFunction()
Dconstant_op_device_assignment.cc45 if (StringAttr device_attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in runOnOperation()
55 StringAttr device_attr = user_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation()
Dmark_ops_for_outside_compilation.cc218 if (parent->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in HasOutsideCompiledAncestor()
231 if (op->hasAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) in MarkVariantInputsOutputs()
244 !input_defining_op->hasAttrOfType<StringAttr>( in MarkVariantInputsOutputs()
248 StringAttr::get(input_defining_op->getContext(), "auto")); in MarkVariantInputsOutputs()
259 !user->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in MarkVariantInputsOutputs()
261 StringAttr::get(user->getContext(), "auto")); in MarkVariantInputsOutputs()
289 StringAttr::get( in MarkUncompilableOps()
307 if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) return; in UnmarkChildren()
311 if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { in UnmarkChildren()
Dcluster_tf_ops_pass.cc67 if (StringAttr attr = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetHost()
126 if (StringAttr attr = func_op.getArgAttrOfType<StringAttr>( in GetFunctionMetadatas()
148 if (StringAttr attr = in GetFunctionMetadatas()
149 defining_op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetFunctionMetadatas()
213 StringAttr::get(context, metadata.input_devices[i])); in CreateFunctions()
218 StringAttr::get(context, metadata.result_devices[i])); in CreateFunctions()
221 func_op->setAttr(kHostAttr, StringAttr::get(context, host)); in CreateFunctions()
Dtpu_device_propagation.cc63 auto lhs_device_attr = lhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
64 auto rhs_device_attr = rhs->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedExecutorOp()
144 func.getArgAttrOfType<StringAttr>(arg.getArgNumber(), kFuncDeviceAttr); in PropagateDevicesFromArguments()
175 auto device_attr = op_to_update->getAttrOfType<StringAttr>(kDeviceAttr); in PropagateDevicesInGraph()
193 mlir::StringAttr::get(op_to_update->getContext(), new_device); in PropagateDevicesInGraph()
218 auto device_attr = func.getResultAttrOfType<StringAttr>( in PropagateDevicesToResults()
222 StringAttr::get(func.getContext(), it->getSecond())); in PropagateDevicesToResults()
Dtpu_compile_op_replication_pass.cc36 if (StringAttr device = op->getAttrOfType<StringAttr>(kDeviceAttr)) { in GetHost()
74 StringAttr::get(&getContext(), device_name)); in runOnOperation()
Dtf_device_replication_pass.cc75 StringAttr device_alias = in runOnOperation()
76 new_op->getAttrOfType<StringAttr>(kDeviceAttr); in runOnOperation()
83 new_op->setAttr(kDeviceAttr, new_devices_array[i].cast<StringAttr>()); in runOnOperation()
Dreadonly_references_to_resources.cc81 StringRef node_name = class_attr.cast<StringAttr>().getValue(); in GetNodeNameFromClassAttr()
141 StringAttr device_attr = in runOnFunction()
142 variable_v2_op->getAttrOfType<StringAttr>("device"); in runOnFunction()
Dtpu_dynamic_layout_pass.cc118 StringAttr device; in IsSupportedInputOp()
120 device = func.getArgAttrOfType<mlir::StringAttr>(arg.getArgNumber(), in IsSupportedInputOp()
123 device = alias.getDefiningOp()->getAttrOfType<StringAttr>(kDeviceAttr); in IsSupportedInputOp()
276 compile->setAttr("metadata", StringAttr::get(compile.getContext(), in HandleCompileAndExecutes()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_types.h41 public llvm::TrailingObjects<TFRTypeStorage, StringAttr> {
42 using KeyTy = ArrayRef<StringAttr>;
48 auto byteSize = TFRTypeStorage::totalSizeToAlloc<StringAttr>(key.size()); in construct()
54 result->getTrailingObjects<StringAttr>()); in construct()
61 return {getTrailingObjects<StringAttr>(), num_attrs}; in GetAttrs()
74 static Derived get(ArrayRef<StringAttr> attrs, MLIRContext* context) { in get()
78 static Derived getChecked(ArrayRef<StringAttr> attrs, Location loc) { in getChecked()
86 Location loc, ArrayRef<StringAttr> attrs) { in verifyConstructionInvariants()
90 ArrayRef<StringAttr> getAttrKeys() { return Base::getImpl()->GetAttrs(); } in getAttrKeys()
Dtfr_ops.cc181 llvm::SmallVector<StringAttr, 4> input_used_attrs, output_used_attrs; in Verify()
217 func.getArgAttrOfType<StringAttr>(arg.index(), kAttrArgumentNameAttr); in Verify()
232 llvm::SmallVector<StringAttr, 4> undefined_attrs; in Verify()
322 [](StringAttr attr) { return attr.getValue().str(); }); in Verify()
586 llvm::SmallVector<StringAttr, 4> attrs; in parseType()
596 attrs.push_back(StringAttr::get(ctx, attr)); in parseType()
621 llvm::ArrayRef<StringAttr> attrs; in printType()
645 [&](StringAttr attr) { os << attr.getValue(); }); in printType()
/external/llvm-project/mlir/include/mlir/Dialect/Utils/
DStructuredOpsUtils.h80 auto strAttr = attr.dyn_cast_or_null<StringAttr>(); in isParallelIterator()
87 auto strAttr = attr.dyn_cast_or_null<StringAttr>(); in isReductionIterator()
94 auto strAttr = attr.dyn_cast_or_null<StringAttr>(); in isWindowIterator()
112 return a.cast<StringAttr>().getValue() == name; in getNumIterators()
139 auto strAttr = attr.dyn_cast_or_null<StringAttr>(); in isSparseDim()
144 auto strAttr = attr.dyn_cast_or_null<StringAttr>(); in isDenseDim()
/external/llvm-project/mlir/test/Bindings/Python/
Dir_attributes.py95 astr = StringAttr(a1)
96 aself = StringAttr(astr)
100 tillegal = StringAttr(Attribute.parse("1.0"))
171 sattr = StringAttr(Attribute.parse('"stringattr"'))
177 print("default_get:", StringAttr.get("foobar"))
179 print("typed_get:", StringAttr.get_typed(
Dir_operation.py224 "foo": StringAttr.get("foo_value"),
225 "bar": StringAttr.get("bar_value"),
402 sattr = StringAttr(op.attributes["dependent"])
561 Operation.create("module", attributes={None:StringAttr.get("name")})
566 Operation.create("module", attributes={42:StringAttr.get("name")})
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dvalidators.h34 auto attr = op->getAttrOfType<StringAttr>("data_format"); in TFDataFormatIsNHWC()
42 auto attr = op->getAttrOfType<StringAttr>("data_format"); in TFDataFormatIsNDHWC()
96 inline bool TFPaddingIsSameOrValid(Operation *op, StringAttr *padding) { in TFPaddingIsSameOrValid()
97 auto padding_attr = op->getAttrOfType<StringAttr>("padding"); in TFPaddingIsSameOrValid()
Dperception_ops_utils.cc74 StringAttr::get(func_.getContext(), kMaxUnpooling)); in RewriteFunc()
125 auto padding = attrs.get("padding").dyn_cast_or_null<StringAttr>(); in CreateCustomOptions()
151 StringAttr::get(func_.getContext(), kImageWarping)); in RewriteFunc()
Dlstm_utils_test.cc81 mlir::StringAttr attr_values = in createLstmCompositeFunc()
129 fused_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
202 EXPECT_EQ(fused_lstm_func_cifg_->getAttrOfType<StringAttr>(kTFImplements) in TEST_F()
227 fused_ln_lstm_func_->getAttrOfType<StringAttr>(kTFImplements).getValue(), in TEST_F()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dprepare_composite_functions_tf.cc77 StringAttr::get(func_.getContext(), "embedding_lookup")); in RewriteFunc()
124 void ConvertTFImplements(FuncOp func, StringAttr attr);
126 void ConvertTFAPIImplements(FuncOp func, StringAttr attr, ModuleOp module);
242 StringAttr attr) { in ConvertTFImplements()
316 StringAttr attr, in ConvertTFAPIImplements()
345 func->getAttrOfType<StringAttr>(kTFImplements); in runOnOperation()
358 func->getAttrOfType<StringAttr>(kTFAPIImplements); in runOnOperation()
/external/llvm-project/flang/include/flang/Lower/
DFIRBuilder.h108 mlir::StringAttr linkage = {},
114 mlir::StringAttr linkage = {});
119 mlir::StringAttr linkage = {},
128 mlir::StringAttr linkage = {}) {
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertKernelFuncToBlob.cpp90 StringAttr translateGPUModuleToBinaryAnnotation(llvm::Module &llvmModule,
146 StringAttr GpuKernelToBlobPass::translateGPUModuleToBinaryAnnotation( in translateGPUModuleToBinaryAnnotation()
151 return StringAttr::get({blob->data(), blob->size()}, loc->getContext()); in translateGPUModuleToBinaryAnnotation()
/external/llvm-project/mlir/lib/IR/
DSymbolTable.cpp29 symbol->getAttrOfType<StringAttr>(SymbolTable::getSymbolAttrName()); in getNameIfSymbol()
34 auto nameAttr = symbol->getAttrOfType<StringAttr>(symbolAttrNameId); in getNameIfSymbol()
194 StringAttr::get(name, symbol->getContext())); in setSymbolName()
200 StringAttr vis = symbol->getAttrOfType<StringAttr>(getVisibilityAttrName()); in getSymbolVisibility()
226 symbol->setAttr(getVisibilityAttrName(), StringAttr::get(visName, ctx)); in setSymbolVisibility()
375 op.getAttrOfType<StringAttr>(mlir::SymbolTable::getSymbolAttrName()); in verifySymbolTable()
404 if (!op->getAttrOfType<StringAttr>(mlir::SymbolTable::getSymbolAttrName())) in verifySymbol()
410 StringAttr visStrAttr = vis.dyn_cast<StringAttr>(); in verifySymbol()
1001 StringAttr visibilityAttr = parser.getBuilder().getStringAttr(visibility); in parseOptionalVisibilityKeyword()
/external/llvm-project/mlir/include/mlir/Bindings/Python/
DAttributes.td29 def : PythonAttr<"::mlir::StringAttr", "_ir.StringAttr">;
/external/llvm-project/debuginfo-tests/llvm-prettyprinters/gdb/
Dmlir-support.cpp38 mlir::Attribute StringAttr = mlir::StringAttr::get("foo", &Context); variable

1234567