| /external/tensorflow/tensorflow/compiler/mlir/xla/ |
| D | type_to_shape_test.cc | 69 EXPECT_EQ(TypeToPrimitiveType(b.getF32Type()), PrimitiveType::F32); in TEST()
80 ShapeUtil::IsScalarWithElementType(TypeToShape(b.getF32Type()), F32)); in TEST()
86 TypeToShape(VectorType::get({8, 128}, b.getF32Type())).ToProto(), in TEST()
102 TypeToShape(MemRefType::get({8, 128}, b.getF32Type())).ToProto(), in TEST()
106 TypeToShape(MemRefType::get({100, 13, 210}, b.getF32Type())).ToProto(), in TEST()
113 VectorType::get({8, 128}, b.getF32Type()))) in TEST()
125 TypeToShape(RankedTensorType::get({8, 128}, b.getF32Type())).ToProto(), in TEST()
131 EXPECT_THAT(TypeToShape(UnrankedTensorType::get(b.getF32Type())).ToProto(), in TEST()
137 {8, 128}, VectorType::get({16, 16}, b.getF32Type()))) in TEST()
161 TypeToShape(RankedTensorType::get({-1, 2, 3}, b.getF32Type()), in TEST()
[all …]
|
| D | hlo_utils.cc | 231 return builder.getF32Type(); in ConvertPrimitiveTypeToMLIRType()
251 return mlir::ComplexType::get(builder.getF32Type()); in ConvertPrimitiveTypeToMLIRType()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/utils/ |
| D | lstm_utils_test.cc | 51 auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); in createLstmCompositeFunc()
52 auto weight_type = RankedTensorType::get(weight_shape, builder->getF32Type()); in createLstmCompositeFunc()
53 auto bias_type = RankedTensorType::get(bias_shape, builder->getF32Type()); in createLstmCompositeFunc()
55 RankedTensorType::get(projection_shape, builder->getF32Type()); in createLstmCompositeFunc()
57 RankedTensorType::get(layer_norm_scale, builder->getF32Type()); in createLstmCompositeFunc()
58 auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); in createLstmCompositeFunc()
|
| D | perception_ops_utils_test.cc | 54 auto input_type = RankedTensorType::get(input_shape, builder->getF32Type()); in createMaxUnpoolingFunc()
56 auto output_type = RankedTensorType::get(output_shape, builder->getF32Type()); in createMaxUnpoolingFunc()
141 auto input_type = RankedTensorType::get({1, 2, 2, 1}, builder_->getF32Type()); in TEST_F()
143 RankedTensorType::get({1, 2, 1, 1}, builder_->getF32Type()); in TEST_F()
|
| D | convert_type.cc | 39 return builder.getF32Type(); in ConvertElementType()
57 return mlir::ComplexType::get(builder.getF32Type()); in ConvertElementType()
|
| D | lstm_utils.cc | 56 auto type = RankedTensorType::get(shape, builder->getF32Type()); in CreateF32SplatConst()
63 auto type = RankedTensorType::get(shape, builder->getF32Type()); in CreatTfF32ConstOp()
64 auto ele_type = RankedTensorType::get({1}, builder->getF32Type()); in CreatTfF32ConstOp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/ |
| D | convert_tensor_test.cc | 47 ConvertTypeToTensorShape(mlir::UnrankedTensorType::get(b.getF32Type())); in TEST()
57 mlir::RankedTensorType::get({-1, 2, 3}, b.getF32Type())); in TEST()
67 mlir::RankedTensorType::get({1, 2, 3}, b.getF32Type())); in TEST()
75 PartialTensorShape output_shape = ConvertTypeToTensorShape(b.getF32Type()); in TEST()
|
| D | convert_attr.cc | 39 return builder->getFloatAttr(builder->getF32Type(), value.f()); in ConvertNonFuncAttributeValue()
58 attrs.push_back(builder->getFloatAttr(builder->getF32Type(), item)); in ConvertNonFuncAttributeValue()
|
| D | convert_type.cc | 40 *type = builder.getF32Type(); in ConvertDataType()
76 *type = mlir::ComplexType::get(builder.getF32Type()); in ConvertDataType()
|
| /external/tensorflow/tensorflow/compiler/mlir/xla/experimental/conv_emitter/ |
| D | conv_emitter.cc | 108 return builder.getF32Type(); in GetShapeInfo()
265 location, mlir::MemRefType::get({}, builder.getF32Type())); in CreateNaiveMlirConv()
270 location, mlir::FloatAttr::get(builder.getF32Type(), 0)), in CreateNaiveMlirConv()
327 builder.getF32Type()); in CreateNaiveMlirConv()
341 builder.getF32Type()); in CreateNaiveMlirConv()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | lower_tf.cc | 896 RankedTensorType::get(padded_shape, rewriter.getF32Type()); in matchAndRewrite()
1399 loc, RankedTensorType::get({}, rewriter.getF32Type()), in_y); in matchAndRewrite()
1401 loc, RankedTensorType::get({}, rewriter.getF32Type()), out_y); in matchAndRewrite()
1404 loc, RankedTensorType::get({}, rewriter.getF32Type()), in_y_f32, in matchAndRewrite()
1408 loc, GetScalarOfType(rewriter.getF32Type(), 0.0)); in matchAndRewrite()
1410 loc, GetScalarOfType(rewriter.getF32Type(), 1.0)); in matchAndRewrite()
1414 RankedTensorType::get({out_height_constant}, rewriter.getF32Type()), in matchAndRewrite()
1419 RankedTensorType::get({out_height_constant}, rewriter.getF32Type()), in matchAndRewrite()
1442 loc, RankedTensorType::get({}, rewriter.getF32Type()), in_x); in matchAndRewrite()
1444 loc, RankedTensorType::get({}, rewriter.getF32Type()), out_x); in matchAndRewrite()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | import_quant_stats_pass.cc | 150 RankedTensorType::get({2}, b.getF32Type()), {min_maxs[0], min_maxs[1]}); in ImportAsStatsOps()
157 RankedTensorType::get(axis_stats_shape, b.getF32Type()), min_maxs); in ImportAsStatsOps()
|
| /external/tensorflow/tensorflow/compiler/mlir/tosa/transforms/ |
| D | legalize_utils.cc | 178 rewriter.getF32Type(), 1.0f, 0, -32768, 32767); in getTosa1DConstTensorTable()
193 auto const_type = RankedTensorType::get({}, rewriter.getF32Type()); in getTosaConstTensorSingleF32()
369 rewriter.getF32Type()); in get1DConstTensor()
|
| /external/llvm-project/mlir/unittests/Dialect/SPIRV/ |
| D | SerializationTest.cpp | 64 llvm::SmallVector<Type, 1> elementTypes{opBuilder.getF32Type()}; in getFloatStructType()
|
| /external/llvm-project/mlir/test/lib/Dialect/Test/ |
| D | TestPatterns.cpp | 292 rewriter.create<ILLegalOpF>(op->getLoc(), rewriter.getF32Type()); in matchAndRewrite()
310 rewriter.create<ILLegalOpF>(op->getLoc(), rewriter.getF32Type()); in matchAndRewrite()
390 rewriter.replaceOpWithNewOp<TestTypeProducerOp>(op, rewriter.getF32Type()); in matchAndRewrite()
881 rewriter.create<ILLegalOpF>(op->getLoc(), rewriter.getF32Type()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/hlo/lib/Dialect/mhlo/transforms/ |
| D | chlo_legalize_to_hlo.cc | 458 rewriter, loc, operands, rewriter.getF32Type(), in matchAndRewrite()
483 rewriter, loc, operands, rewriter.getF32Type(), in matchAndRewrite()
947 FloatType min_precision_ty = rewriter.getF32Type(); in matchAndRewrite()
960 FloatType min_precision_ty = rewriter.getF32Type(); in matchAndRewrite()
974 FloatType min_precision_ty = rewriter.getF32Type(); in matchAndRewrite()
988 FloatType min_precision_ty = rewriter.getF32Type(); in matchAndRewrite()
|
| D | legalize_trigonometric_to_approximation.cc | 57 arg = rewriter.create<FPExtOp>(loc, arg, rewriter.getF32Type()); in matchAndRewrite()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | default_quant_params.cc | 221 builder.getF32Type(), is_signed_); in GetDefaultQuantParams()
|
| D | prepare_tf.cc | 1019 epsilon = rewriter.getFloatAttr(rewriter.getF32Type(), 0.0001f); in matchAndRewrite()
1036 rewriter.getFloatAttr(rewriter.getF32Type(), 1.0f); in matchAndRewrite()
1155 ::mlir::RankedTensorType::get({1}, rewriter.getF32Type()); in matchAndRewrite()
|
| /external/llvm-project/mlir/lib/IR/ |
| D | Builders.cpp | 49 FloatType Builder::getF32Type() { return FloatType::getF32(context); } in getF32Type() function in Builder
183 return FloatAttr::get(getF32Type(), APFloat(value)); in getF32FloatAttr()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/ |
| D | flatbuffer_import.cc | 171 flags, storage_type, builder.getF32Type(), scales, in GetQuantizedType()
176 flags, storage_type, builder.getF32Type(), quant_params.scale.at(0), in GetQuantizedType()
263 mlir::RankedTensorType::get({2}, b.getF32Type()), in ConvertMinMaxToStatsOp()
271 mlir::RankedTensorType::get(axis_stats_shape, b.getF32Type()), in ConvertMinMaxToStatsOp()
|
| /external/llvm-project/mlir/unittests/TableGen/ |
| D | OpBuildGen.cpp | 38 i32Ty(builder.getI32Type()), f32Ty(builder.getF32Type()), in OpBuildGenTest()
|
| /external/llvm-project/mlir/lib/Parser/ |
| D | TypeParser.cpp | 353 return builder.getF32Type(); in parseNonFunctionType()
|
| /external/llvm-project/mlir/include/mlir/IR/ |
| D | Builders.h | 67 FloatType getF32Type();
|
| /external/tensorflow/tensorflow/compiler/mlir/tfr/passes/ |
| D | raise_to_tf.cc | 141 result_elt_type = rewriter.getF32Type(); in CastToNonDerivedType()
|