Home
last modified time | relevance | path

Searched refs:getI32IntegerAttr (Results 1 – 25 of 66) sorted by relevance

123

/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dvalidators.cc44 *x = b.getI32IntegerAttr(elements[1].cast<IntegerAttr>().getInt()); in TFIntListIs1XY1()
45 *y = b.getI32IntegerAttr(elements[2].cast<IntegerAttr>().getInt()); in TFIntListIs1XY1()
84 *x = b.getI32IntegerAttr(elements[1].cast<IntegerAttr>().getInt()); in TFIntListIs1XYZ1()
85 *y = b.getI32IntegerAttr(elements[2].cast<IntegerAttr>().getInt()); in TFIntListIs1XYZ1()
86 *z = b.getI32IntegerAttr(elements[3].cast<IntegerAttr>().getInt()); in TFIntListIs1XYZ1()
/external/llvm-project/mlir/lib/Dialect/Tosa/Utils/
DQuantUtils.cpp137 builder.getI32IntegerAttr(inputZp), builder.getI32IntegerAttr(weightZp), in buildConvOpQuantizationAttr()
173 builder.getI32IntegerAttr(aZp), builder.getI32IntegerAttr(bZp), in buildMatMulOpQuantizationAttr()
209 builder.getI32IntegerAttr(inputZp), builder.getI32IntegerAttr(outputZp), in buildUnaryOpQuantizationAttr()
235 builder.getI32IntegerAttr(inputZp), builder.getContext()); in buildPadOpQuantizationAttr()
/external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/
Dconvert_tfl_uint8.cc107 builder.getI32IntegerAttr( in matchAndRewrite()
181 builder.getI32IntegerAttr( in convert_graph_uint8_tensor()
194 builder.getI32IntegerAttr(uint8_zp), in convert_graph_uint8_tensor()
195 builder.getI32IntegerAttr(int8_zp), in convert_graph_uint8_tensor()
250 builder.getI32IntegerAttr( in convert_graph_uint8_tensor()
317 builder.getI32IntegerAttr(int8_zp), in convert_graph_uint8_tensor()
318 builder.getI32IntegerAttr(uint8_zp), in convert_graph_uint8_tensor()
Dlegalize_utils.cc40 rewriter.getI32IntegerAttr(static_cast<int32_t>(input_zp)), in buildRescale()
41 rewriter.getI32IntegerAttr(static_cast<int32_t>(output_zp)), in buildRescale()
108 op->getLoc(), output_type, conv_val, rewriter.getI32IntegerAttr(0), in buildRescaleOpConvOutput()
109 rewriter.getI32IntegerAttr(output_zp), in buildRescaleOpConvOutput()
149 op->getLoc(), output_type, conv_val, rewriter.getI32IntegerAttr(0), in buildRescaleOpConvOutput()
150 rewriter.getI32IntegerAttr(output_zp), in buildRescaleOpConvOutput()
/external/llvm-project/mlir/lib/Dialect/SPIRV/
DSPIRVAttributes.cpp120 auto descriptorSetAttr = b.getI32IntegerAttr(descriptorSet); in get()
121 auto bindingAttr = b.getI32IntegerAttr(binding); in get()
123 storageClass ? b.getI32IntegerAttr(static_cast<uint32_t>(*storageClass)) in get()
187 auto versionAttr = b.getI32IntegerAttr(static_cast<uint32_t>(version)); in get()
192 capAttrs.push_back(b.getI32IntegerAttr(static_cast<uint32_t>(cap))); in get()
DSPIRVDialect.cpp891 descriptorSetAttr = builder.getI32IntegerAttr(descriptorSet); in parseInterfaceVarABIAttr()
907 bindingAttr = builder.getI32IntegerAttr(binding); in parseInterfaceVarABIAttr()
923 storageClassAttr = builder.getI32IntegerAttr( in parseInterfaceVarABIAttr()
954 builder.getI32IntegerAttr(static_cast<uint32_t>(*versionSymbol)); in parseVerCapExtAttr()
970 builder.getI32IntegerAttr(static_cast<uint32_t>(*capSymbol))); in parseVerCapExtAttr()
DSPIRVOps.cpp146 state.addAttribute(attrName, parser.getBuilder().getI32IntegerAttr( in parseEnumStrAttr()
162 state.addAttribute(attrName, parser.getBuilder().getI32IntegerAttr( in parseEnumKeywordAttr()
182 state.addAttribute(attrName, builder.getI32IntegerAttr(0)); in parseControlAttribute()
1609 builder.getI32IntegerAttr(static_cast<int32_t>(executionModel)), in build()
1671 builder.getI32IntegerAttr(static_cast<int32_t>(executionMode)), in build()
1828 builder.getI32IntegerAttr(static_cast<uint32_t>(control))); in build()
1919 builder.getI32IntegerAttr(descriptorSet)); in build()
1922 builder.getI32IntegerAttr(binding)); in build()
2252 builder.getI32IntegerAttr( in build()
2441 builder.getI32IntegerAttr(static_cast<int32_t>(addressingModel))); in build()
[all …]
/external/llvm-project/mlir/test/lib/Dialect/Tosa/
DTosaTestPasses.cpp80 rewriter.getI32IntegerAttr( in matchAndRewrite()
169 rewriter.getI32IntegerAttr(0), rewriter.getI32IntegerAttr(outputZp), in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/lite/transforms/
Dlegalize_tf.cc293 auto values_count = rewriter.getI32IntegerAttr(tf_pack_op.N()); in matchAndRewrite()
295 auto axis = rewriter.getI32IntegerAttr(tf_pack_op.axis()); in matchAndRewrite()
307 auto num_split = rewriter.getI32IntegerAttr(tf_split_op.num_split()); in matchAndRewrite()
320 auto num_split = rewriter.getI32IntegerAttr(tf_splitv_op.num_split()); in matchAndRewrite()
333 auto num = rewriter.getI32IntegerAttr(tf_unpack_op.num()); in matchAndRewrite()
335 auto axis = rewriter.getI32IntegerAttr(tf_unpack_op.axis()); in matchAndRewrite()
359 dilation_depth_factor = rewriter.getI32IntegerAttr(1); in matchAndRewrite()
360 dilation_height_factor = rewriter.getI32IntegerAttr(1); in matchAndRewrite()
361 dilation_width_factor = rewriter.getI32IntegerAttr(1); in matchAndRewrite()
Ddense_to_sparse.cc255 builder->getI32IntegerAttr(metadata[2 * i][0]), in BuildSparsityParameterAttribute()
260 builder->getStringAttr("SPARSE_CSR"), builder->getI32IntegerAttr(0), in BuildSparsityParameterAttribute()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dtranslate_utils.cc26 b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer())); in PopulateTfVersions()
28 "min_consumer", b.getI32IntegerAttr(versions.min_consumer())); in PopulateTfVersions()
Ddevice_util.cc54 builder->getI32IntegerAttr(std::stoi(cc[1].str())), in ParseGpuDeviceMetadata()
55 builder->getI32IntegerAttr(std::stoi(cc[2].str())), in ParseGpuDeviceMetadata()
Ddevice_util_test.cc188 gpu0, mlir::TF::GpuDeviceMetadata::get(builder.getI32IntegerAttr(1), in TEST()
189 builder.getI32IntegerAttr(2), in TEST()
/external/tensorflow/tensorflow/compiler/mlir/tools/kernel_gen/transforms/
Dtf_kernel_to_llvm_pass.cc111 builder.getI32IntegerAttr(1)); in generateParamsArray()
115 loc, llvm_int32_type_, builder.getI32IntegerAttr(num_arguments)); in generateParamsArray()
119 builder.getI32IntegerAttr(0)); in generateParamsArray()
122 loc, llvm_int32_type_, builder.getI32IntegerAttr(en.index())); in generateParamsArray()
Dtf_framework_legalize_to_llvm.cc97 rewriter.getI32IntegerAttr(tf_alloc_op.output_index().hasValue() in matchAndRewrite()
188 loc, llvm_i32_type, rewriter->getI32IntegerAttr(0)); in ConvertI32ArrayAttrToStackAllocatedArray()
196 loc, llvm_i32_type, rewriter->getI32IntegerAttr(array_attr.size())); in ConvertI32ArrayAttrToStackAllocatedArray()
202 loc, llvm_i32_type, rewriter->getI32IntegerAttr(dim.index())); in ConvertI32ArrayAttrToStackAllocatedArray()
207 rewriter->getI32IntegerAttr( in ConvertI32ArrayAttrToStackAllocatedArray()
/external/llvm-project/mlir/unittests/Dialect/SPIRV/
DSerializationTest.cpp49 builder.getI32IntegerAttr(static_cast<uint32_t>( in createModuleOp()
52 builder.getI32IntegerAttr( in createModuleOp()
/external/llvm-project/mlir/lib/Conversion/GPUCommon/
DConvertLaunchFuncToRuntimeCalls.cpp499 builder.getI32IntegerAttr(1)); in generateParamsArray()
503 loc, llvmInt32Type, builder.getI32IntegerAttr(numArguments)); in generateParamsArray()
507 builder.getI32IntegerAttr(0)); in generateParamsArray()
510 loc, llvmInt32Type, builder.getI32IntegerAttr(en.index())); in generateParamsArray()
612 rewriter.getI32IntegerAttr(0)); in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Ddevice_index_selector.cc69 DenseElementsAttr::get(type, b.getI32IntegerAttr(index)); in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/GPUToVulkan/
DConvertLaunchFuncToVulkanCalls.cpp219 loc, getInt32Type(), builder.getI32IntegerAttr(0)); in createBindMemRefCalls()
226 loc, getInt32Type(), builder.getI32IntegerAttr(en.index())); in createBindMemRefCalls()
390 builder.getI32IntegerAttr(spirvAttributes.first.getValue().size())); in translateVulkanLaunchCall()
/external/llvm-project/mlir/lib/Dialect/SPIRV/Serialization/
DDeserializer.cpp735 opBuilder.getI32IntegerAttr(llvm::bit_cast<int32_t>(operands.front()))); in processMemoryModel()
738 opBuilder.getI32IntegerAttr(llvm::bit_cast<int32_t>(operands.back()))); in processMemoryModel()
766 symbol, opBuilder.getI32IntegerAttr(static_cast<int32_t>(words[2]))); in processDecoration()
809 symbol, opBuilder.getI32IntegerAttr(static_cast<int32_t>(words[2]))); in processDecoration()
1984 auto control = builder.getI32IntegerAttr(selectionControl); in createSelectionOp()
1996 auto control = builder.getI32IntegerAttr(loopControl); in createLoopOp()
2550 auto execModel = opBuilder.getI32IntegerAttr(words[wordIndex++]); in processOp()
2602 auto execMode = opBuilder.getI32IntegerAttr(words[wordIndex++]); in processOp()
2607 attrListElems.push_back(opBuilder.getI32IntegerAttr(words[wordIndex++])); in processOp()
2744 "memory_access", opBuilder.getI32IntegerAttr(attrValue))); in processOp()
[all …]
/external/tensorflow/tensorflow/compiler/mlir/
Dmlir_graph_optimization_pass_test.cc59 auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(0)); in Run()
60 auto min_consumer = b.getNamedAttr("min_consumer", b.getI32IntegerAttr(0)); in Run()
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_operator.cc194 return builder.getI32IntegerAttr(value); in BuildI32Attr()
205 return builder.getI32IntegerAttr(value); in BuildPositiveI32Attr()
/external/llvm-project/mlir/test/lib/Transforms/
DTestDecomposeCallGraphTypes.cpp61 loc, resultType.getType(i), value, builder.getI32IntegerAttr(i)); in runOnOperation()
/external/llvm-project/mlir/lib/Conversion/SCFToSPIRV/
DSCFToSPIRV.cpp132 auto loopControl = rewriter.getI32IntegerAttr( in matchAndRewrite()
217 auto selectionControl = rewriter.getI32IntegerAttr( in matchAndRewrite()
/external/tensorflow/tensorflow/compiler/mlir/xla/ir/
Dmlir_hlo_builder.cc282 loc_, ty, GetValue(operand), builder_.getI32IntegerAttr(exponent_bits), in ReducePrecisionInternal()
283 builder_.getI32IntegerAttr(mantissa_bits)); in ReducePrecisionInternal()
359 loc_, ty, builder_.getI32IntegerAttr(algorithm), GetValue(initial_state)); in RngBitGeneratorInternal()
505 builder_.getI32IntegerAttr(index))); in GetTupleElementInternal()

123