Searched refs:isBF16 (Results 1 – 21 of 21) sorted by relevance
79 return tensorType.getElementType().isBF16(); in TFTypeIsBFloat16Tensor()
38 } else if (element_type.isBF16()) { in CreateConstOpWithSingleValue()
25 bool Type::isBF16() const { return isa<BFloat16Type>(); } in isBF16() function in Type
30 size_t bit_width = new_type.isBF16() ? 64 : new_type.getIntOrFloatBitWidth(); in ConvertElementsAttr()
225 if (!matmul.T().isF32() && !matmul.T().isBF16()) { in AreFuseCompatible()
1218 assert(element_type.isBF16()); in matchAndRewrite()
180 using Type::isBF16;
117 bool isBF16() const;
459 def BF16 : Type<CPred<"$_self.isBF16()">, "bfloat16 type">,
103 } else if (type.isBF16()) { in ConvertScalarTypeToDataType()
46 if (type.isBF16()) { in TypeToPrimitiveType()
168 if (element_type.isBF16()) { in CopyDenseElementsDataToXlaFormat()
79 if (elemType.isBF16() || elemType.isF32()) return true; in IsCompatibleTypeWithTFLCastOp()
864 return element_type.isBF16() || element_type.isF32(); in CanFuseAffineOp()
70 bool mlirTypeIsABF16(MlirType type) { return unwrap(type).isBF16(); } in mlirTypeIsABF16()
112 } else if (type.isBF16()) { in get()
793 if (type.isBF16() || type.isF16() || type.isF32()) in runOnOperation()
174 if (type.isBF16()) { in parseAndVerifyType()
568 bool ScalarType::isValid(FloatType type) { return !type.isBF16(); } in isValid()
335 return shaped_type.getElementType().isBF16(); in IsBF16ShapedType()1137 if (getType().cast<ShapedType>().getElementType().isBF16()) { in fold()
186 if (input_type.isBF16() || input_type.isF16()) return FloatType::getF32(ctx); in GetSumAccumulationType()900 return (ty.isF16() || ty.isBF16()) ? FloatType::getF32(ty.getContext()) : ty; in GetAccumulationType()914 } else if (element_ty.isBF16()) { in GetEpsilonValue()1312 if (!element_type.isBF16()) return failure(); in matchAndRewrite()