Searched refs:getF16Type (Results 1 – 9 of 9) sorted by relevance
106 return builder.getF16Type(); in GetShapeInfo()252 CHECK(input_shape_info.element_type == builder.getF16Type()); in CreateNaiveMlirConv()253 CHECK(filter_shape_info.element_type == builder.getF16Type()); in CreateNaiveMlirConv()254 CHECK(output_shape_info.element_type == builder.getF16Type()); in CreateNaiveMlirConv()365 builder.getF16Type()), in CreateNaiveMlirConv()
37 return builder.getF16Type(); in ConvertElementType()
70 result = rewriter.create<FPTruncOp>(loc, result, rewriter.getF16Type()); in matchAndRewrite()
37 *type = builder.getF16Type(); in ConvertDataType()
47 FloatType Builder::getF16Type() { return FloatType::getF16(context); } in getF16Type() function in Builder187 return FloatAttr::get(getF16Type(), value); in getF16FloatAttr()
350 return builder.getF16Type(); in parseNonFunctionType()
66 FloatType getF16Type();
227 return builder.getF16Type(); in ConvertPrimitiveTypeToMLIRType()
1193 floatTy = opBuilder.getF16Type(); in processType()