Searched refs:squeeze_v2 (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_dispatch_test.py | 651 op=array_ops.squeeze_v2,
|
| D | ragged_dispatch.py | 498 (array_ops.squeeze_v2, ragged_squeeze_op.squeeze, ['input']),
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | cond_v2.py | 81 pred = array_ops.squeeze_v2(pred)
|
| D | while_v2.py | 135 pred = array_ops.squeeze_v2(pred)
|
| D | array_ops.py | 4476 def squeeze_v2(input, axis=None, name=None): function
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | functional.py | 625 tensor = array_ops.squeeze_v2(tensor, axis=-1)
|
| /external/tensorflow/tensorflow/python/kernel_tests/ |
| D | cond_v2_test.py | 64 array_ops.squeeze_v2(pred), true_fn, false_fn, name="expected")
|