code: def _prepare_pde_coeff(raw_coeff, value_grid):
      # Converts values received from second_order_coeff_fn and similar Callables
      # into a format usable further down in the pipeline.
      if raw_coeff is None:
          return None
      dtype = value_grid.dtype
      coeff = tf.convert_to_tensor(raw_coeff, dtype=dtype)
      coeff = tf.broadcast_to(coeff, utils.get_shape(value_grid))
      return coeff
  
imports: import tensorflow.compat.v2
  from tf_quant_finance import utils
label_desc: Use image node_id 15 for calling a global function with example usage: _prepare_pde_coeff(raw_coeff, value_grid) and returns: coeff, None