code: def _prepare_pde_coeffs(raw_coeffs, value_grid):
      """Prepares values received from second_order_coeff_fn and similar."""
      if raw_coeffs is None:
          return None
      dtype = value_grid.dtype
      coeffs = tf.convert_to_tensor(raw_coeffs, dtype=dtype)
      broadcast_shape = utils.get_shape(value_grid)
      coeffs = tf.broadcast_to(coeffs, broadcast_shape)
      return coeffs
  
file_path: tf-quant-finance/tf_quant_finance/math/pde/steppers/parabolic_equation_stepper.py
imports: import tensorflow.compat.v2
  from tf_quant_finance import utils
label_desc: Use image node_id 8 for calling a global function with example usage: _prepare_pde_coeffs(raw_coeffs, value_grid) and returns: coeffs, None