Searched defs:outer_axis (Results 1 – 5 of 5) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/ |
| H A D | ragged_merge_dims_op_test.py | 180 outer_axis, argument
257 outer_axis, argument
|
| H A D | ragged_tensor.py | 1464 def merge_dims(self, outer_axis, inner_axis): argument
2866 def merge_dims(value, outer_axis, inner_axis): argument
|
| H A D | dynamic_ragged_shape_test.py | 2728 def test_merge_dims(self, lengths, num_row_partitions, outer_axis, inner_axis, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/structured/ |
| H A D | structured_tensor.py | 1075 def merge_dims(self, outer_axis, inner_axis): argument
1553 def _merge_dims(value, outer_axis, inner_axis): argument
|
| H A D | structured_tensor_test.py | 1262 def testMergeDims(self, st, outer_axis, inner_axis, expected): argument
|