code: def testV2PredBroadcasting(self):
      pred = np.array([1, 0, 0]).reshape((3, 1))
      x = np.random.randn(3, 4)
      y = np.random.randn(3, 4)
      np_val = np.where(pred, x, y)
      with self.test_session():
          tf_val = self.evaluate(array_ops.where_v2(pred, x, y))
      self.assertAllClose(tf_val, np_val)
  
imports: import itertools
  import sys
  import numpy
  from tensorflow.python.client import session
  from tensorflow.python.framework import constant_op
  from tensorflow.python.framework import dtypes
  from tensorflow.python.framework import ops
  from tensorflow.python.framework import test_util
  from tensorflow.python.ops import array_ops
  from tensorflow.python.ops import random_ops
  from tensorflow.python.ops import resource_variable_ops
  from tensorflow.python.platform import benchmark
  from tensorflow.python.platform import test
label_desc: Use image node_id 44 for calling the WhereOpTest obj's underlying member method code with example usage: obj.testV2PredBroadcasting() without return types