Searched refs:true_vector (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/distributions/ |
| D | util.py | 660 def pick_vector(cond, true_vector, false_vector, name="pick_vector"): argument
686 with ops.name_scope(name, values=(cond, true_vector, false_vector)):
693 return true_vector if cond_value_static else false_vector
694 true_vector = ops.convert_to_tensor(true_vector, name="true_vector")
696 if true_vector.dtype != false_vector.dtype:
699 (true_vector, true_vector.dtype, false_vector, false_vector.dtype))
700 n = array_ops.shape(true_vector)[0]
702 array_ops.concat([true_vector, false_vector], 0),
|