Home
last modified time | relevance | path

Searched refs:getBF16 (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/python/mlir_wrapper/
Dtypes.cc30 .def("getBF16", &mlir::FloatType::getBF16) in init_types()
/external/tensorflow/tensorflow/compiler/mlir/lite/utils/
Dconstant_utils.cc39 auto floatType = mlir::FloatType::getBF16(element_type.getContext()); in CreateConstOpWithSingleValue()
/external/llvm-project/mlir/include/mlir/IR/
DBuiltinTypes.h166 static FloatType getBF16(MLIRContext *ctx);
193 inline FloatType FloatType::getBF16(MLIRContext *ctx) { in getBF16() function
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dconvert_tensor_test.cc132 mlir::FloatType::getBF16(&context))); in TEST_F()
/external/llvm-project/mlir/unittests/IR/
DAttributeTest.cpp145 FloatType floatTy = FloatType::getBF16(&context); in TEST()
/external/llvm-project/mlir/lib/CAPI/IR/
DBuiltinTypes.cpp73 return wrap(FloatType::getBF16(unwrap(ctx))); in mlirBF16TypeGet()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/ir/
Dtf_types.cc157 if (isa<Bfloat16RefType>()) return mlir::FloatType::getBF16(ctx); in RemoveRef()
/external/llvm-project/mlir/lib/IR/
DBuilders.cpp45 FloatType Builder::getBF16Type() { return FloatType::getBF16(context); } in getBF16Type()
/external/llvm-project/flang/lib/Lower/
DConvertType.cpp71 return mlir::FloatType::getBF16(context); in genFIRType()