code: def test_bf16_square():
      dshapes = [(16,), (3, 16), (3, 16, 16), (3, 16, 16, 16)]
      for data_shape in dshapes:
          data_sym_fp32 = mx.sym.Variable(name="data")
          data_sym_bf16 = mx.sym.Variable(name="data", dtype="bfloat16")
          sym_bf16 = mx.sym.square(data_sym_bf16)
          sym_fp32 = mx.sym.square(data_sym_fp32)
  
          check_operator_accuracy(
              sym_fp32, sym_bf16, data_shape, bf16_use_fp32_params=True
          )
  
imports: import os
  import sys
  import mxnet
  import numpy
  from random import randint
  import itertools
  from mxnet.test_utils import assert_almost_equal_with_err, rand_shape_nd
  import pytest
label_desc: Use image node_id 13 for calling a global function with example usage: test_bf16_square() without return types