Function Bfloat16Dtype

tensorflow/compiler/xla/python/bfloat16.cc:1542–1551

}

StatusOr<py::object> Bfloat16Dtype() {
  if (npy_bfloat16 < 0) {
    // Not yet initialized. We assume the GIL protects npy_bfloat16.
    if (!Initialize()) {
      return InternalError("Bfloat16 numpy type initialization failed.");
    }
  }
  return py::object(reinterpret_cast<PyObject*>(&PyBfloat16_Type),
                    /*is_borrowed=*/true);
}

}  // namespace xla