Home
last modified time | relevance | path

Searched refs:getArrayAttr (Results 1 – 25 of 32) sorted by relevance

12

/external/llvm-project/mlir/lib/IR/
DBuilders.cpp202 ArrayAttr Builder::getArrayAttr(ArrayRef<Attribute> value) { in getArrayAttr() function in Builder
224 return getArrayAttr(attrs); in getBoolArrayAttr()
230 return getArrayAttr(attrs); in getI32ArrayAttr()
235 return getArrayAttr(attrs); in getI64ArrayAttr()
243 return getArrayAttr(attrs); in getIndexArrayAttr()
249 return getArrayAttr(attrs); in getF32ArrayAttr()
255 return getArrayAttr(attrs); in getF64ArrayAttr()
261 return getArrayAttr(attrs); in getStrArrayAttr()
267 return getArrayAttr(attrs); in getTypeArrayAttr()
273 return getArrayAttr(attrs); in getAffineMapArrayAttr()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Ddense_to_sparse.cc256 builder->getArrayAttr({}), builder->getArrayAttr({}), in BuildSparsityParameterAttribute()
267 builder->getI32ArrayAttr(b_map), builder->getArrayAttr(dim_metadata), in BuildSparsityParameterAttribute()
Dinitialize_variables.cc151 builder.getArrayAttr(builder.getSymbolRefAttr(kSessionInitFuncName))); in CreateSessionInitFunc()
/external/llvm-project/mlir/lib/Dialect/PDLInterp/IR/
DPDLInterp.cpp69 state.addAttribute("attributeNames", builder.getArrayAttr(attrNames)); in parseCreateOperationOp()
/external/llvm-project/mlir/lib/Dialect/SPIRV/
DSPIRVAttributes.cpp199 return get(versionAttr, b.getArrayAttr(capAttrs), b.getArrayAttr(extAttrs)); in get()
DSPIRVDialect.cpp981 capabilitiesAttr = builder.getArrayAttr(capabilities); in parseVerCapExtAttr()
1003 extensionsAttr = builder.getArrayAttr(extensions); in parseVerCapExtAttr()
DSPIRVOps.cpp1214 builder.getArrayAttr({trueWeight, falseWeight})); in parseBranchConditionalOp()
1611 builder.getArrayAttr(interfaceVars)); in build()
1641 parser.getBuilder().getArrayAttr(interfaceVars)); in parseEntryPointOp()
3342 parser.getBuilder().getArrayAttr(constituents)); in parseSpecConstantCompositeOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_attr.cc79 return builder->getArrayAttr( in ConvertNonFuncAttributeValue()
Dtpu_rewrite_device_util_test.cc656 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
676 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
744 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
799 cluster->setAttr(kDeviceAssignmentAttr, builder.getArrayAttr({})); in TEST()
/external/tensorflow/tensorflow/compiler/mlir/xla/
Dattribute_importer.cc44 return builder->getArrayAttr(operand_precision_attrs); in ConvertPrecisionConfig()
Dhlo_function_importer.cc395 mlir::ArrayAttr layout = builder_->getArrayAttr(array_ref); in ImportInstructionImpl()
/external/llvm-project/mlir/lib/Interfaces/
DViewLikeInterface.cpp114 result.addAttribute(attrName, parser.getBuilder().getArrayAttr({})); in parseListOfOperandsOrIntegers()
/external/llvm-project/mlir/lib/Dialect/PDL/IR/
DPDL.cpp166 state.addAttribute("attributeNames", builder.getArrayAttr(attrNames)); in parseOperationOp()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_saved_model.cc487 op->setAttr("initializers", rewriter.getArrayAttr(to_keep)); in matchAndRewrite()
/external/llvm-project/mlir/lib/Dialect/Linalg/Transforms/
DFusionOnTensors.cpp226 rewriter.getArrayAttr(fusedIndexMaps), in fuseTensorOpsImpl()
239 rewriter.getArrayAttr(fusedIndexMaps), in fuseTensorOpsImpl()
/external/llvm-project/mlir/include/mlir/IR/
DBuilders.h97 ArrayAttr getArrayAttr(ArrayRef<Attribute> value);
/external/tensorflow/tensorflow/compiler/mlir/tfr/passes/
Draise_to_tf.cc217 {attr_names[0].getValue(), rewriter.getArrayAttr(input_types)}); in AddDerivedAttrs()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/
DDeserializer.cpp1678 auto attr = opBuilder.getArrayAttr(elements); in processConstantComposite()
1717 opBuilder.getArrayAttr(elements)); in processSpecConstantComposite()
2580 opBuilder.getArrayAttr(interface)); in processOp()
2609 auto values = opBuilder.getArrayAttr(attrListElems); in processOp()
/external/tensorflow/tensorflow/compiler/mlir/tfr/ir/
Dtfr_ops.cc508 auto array_attr = rewriter.getArrayAttr(array_list); in matchAndRewrite()
/external/llvm-project/mlir/examples/toy/Ch7/mlir/
DMLIRGen.cpp441 mlir::ArrayAttr dataAttr = builder.getArrayAttr(attrElements); in getConstantAttr()
/external/llvm-project/flang/lib/Optimizer/Dialect/
DFIROps.cpp1133 parser.getBuilder().getArrayAttr(attrs)); in parseSelectCase()
1173 result.addAttribute(getCasesAttr(), builder.getArrayAttr(compareAttrs)); in build()
1333 bld.getArrayAttr(attrs)); in parseSelectType()
/external/llvm-project/mlir/lib/Parser/
DAttributeParser.cpp79 return builder.getArrayAttr(elements); in parseAttribute()
/external/llvm-project/mlir/include/mlir/Dialect/PDL/IR/
DPDLOps.td65 params.empty() ? ArrayAttr() : $_builder.getArrayAttr(params));
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/
Dimport_model.cc1204 return builder_.getArrayAttr( in ConvertAttributeValue()
2902 leaf_index_paths_.push_back(builder_.getArrayAttr(current_index_path_)); in RecursivelyFindLeaves()
3784 module_->getLoc(), builder.getArrayAttr(init_sym_refs)); in ConvertSignatures()
/external/llvm-project/mlir/lib/Dialect/Vector/
DVectorTransforms.cpp2038 rewriter.getArrayAttr(adjustIter(op.iterator_types(), iterIndex)); in lowerParallel()
2089 rewriter.getArrayAttr(adjustIter(op.iterator_types(), iterIndex)); in lowerReduction()

12