code: def test_bf16_to_float32():
      x = torch.randn([2, 2], dtype=torch.bfloat16)
      converted_x = _bf16_to_float32(x)
      assert (
          x.dtype == torch.bfloat16
          and converted_x.dtype == torch.float32
          and torch.equal(converted_x, x.float())
      )
  
file_path: doctr/tests/pytorch/test_models_utils_pt.py
imports: import os
  import pytest
  import torch
  from torch import nn
  from doctr.models.utils import _bf16_to_float32, _copy_tensor, conv_sequence_pt, load_pretrained_params, set_device_and_dtype
label_desc: Use image node_id 2 for calling a global function with example usage: test_bf16_to_float32() without return types