Searched refs:squeeze_v2 (Results 1 – 7 of 7) sorted by relevance
29 @dispatch.dispatch_for_api(array_ops.squeeze_v2)
988 op=array_ops.squeeze_v2,
78 pred = array_ops.squeeze_v2(pred)
149 pred = array_ops.squeeze_v2(pred)
4604 def squeeze_v2(input, axis=None, name=None): function
617 tensor = array_ops.squeeze_v2(tensor, axis=-1)
60 array_ops.squeeze_v2(pred), true_fn, false_fn, name="expected")