| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_merge_dims_op_test.py | 180 outer_axis, argument
188 actual = x.merge_dims(outer_axis, inner_axis)
190 if outer_axis >= 0 and inner_axis >= 0:
192 x.shape.rank - (inner_axis - outer_axis))
195 if outer_axis >= 0 and inner_axis >= 0:
196 actual_with_neg_axis = x.merge_dims(outer_axis - x.shape.rank,
201 if (not context.executing_eagerly() and outer_axis >= 0 and
208 outer_axis, inner_axis)
257 outer_axis, argument
264 self.evaluate(x.merge_dims(outer_axis, inner_axis))
|
| D | dynamic_ragged_shape.py | 875 def _merge_dims(self, outer_axis: int,
908 outer_axis = array_ops.get_positive_axis(
909 outer_axis,
918 if not outer_axis <= inner_axis:
921 if outer_axis == inner_axis:
926 self._inner_shape, self._static_inner_shape, outer_axis, inner_axis)
934 if outer_axis == 0:
941 prefix_rp = self._row_partitions[:outer_axis - 1]
943 internal_rp = self._row_partitions[outer_axis - 1:inner_axis]
949 elif outer_axis > self.num_row_partitions:
[all …]
|
| D | ragged_tensor.py | 1464 def merge_dims(self, outer_axis, inner_axis): argument
1497 outer_axis = array_ops.get_positive_axis(
1498 outer_axis,
1507 if not outer_axis <= inner_axis:
1510 return merge_dims(self, outer_axis, inner_axis)
2866 def merge_dims(value, outer_axis, inner_axis): argument
2880 if outer_axis == inner_axis:
2884 while outer_axis == 0 and isinstance(value, RaggedTensor):
2894 new_shape = old_shape[:outer_axis] + [-1] + old_shape[inner_axis + 1:]
2898 [old_shape[:outer_axis], [-1], old_shape[inner_axis + 1:]], axis=0)
[all …]
|
| D | dynamic_ragged_shape_test.py | 2675 outer_axis=1,
2683 outer_axis=1,
2691 outer_axis=0,
2698 outer_axis=1,
2706 outer_axis=2,
2716 outer_axis=0,
2723 outer_axis=1,
2728 def test_merge_dims(self, lengths, num_row_partitions, outer_axis, inner_axis, argument
2732 actual = original._merge_dims(outer_axis, inner_axis)
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 1075 def merge_dims(self, outer_axis, inner_axis): argument
1101 outer_axis = array_ops.get_positive_axis(
1102 outer_axis,
1111 if not outer_axis <= inner_axis:
1113 'inner_axis (%d)' % (outer_axis, inner_axis))
1114 return _merge_dims(self, outer_axis, inner_axis)
1553 def _merge_dims(value, outer_axis, inner_axis): argument
1555 assert outer_axis < inner_axis
1557 return ragged_tensor.merge_dims(value, outer_axis, inner_axis)
1560 fields = dict((k, _merge_dims(v, outer_axis, inner_axis))
[all …]
|
| D | structured_tensor_test.py | 1202 outer_axis=0, inner_axis=1,
1207 outer_axis=0, inner_axis=1,
1212 outer_axis=1, inner_axis=2,
1217 outer_axis=0, inner_axis=2,
1223 outer_axis=0, inner_axis=1,
1230 outer_axis=0, inner_axis=2,
1237 outer_axis=0, inner_axis=3,
1244 outer_axis=1, inner_axis=2,
1251 outer_axis=1, inner_axis=3,
1258 outer_axis=2, inner_axis=3,
[all …]
|
| /external/tensorflow/tensorflow/tools/api/golden/v2/ |
| D | tensorflow.-ragged-tensor.pbtxt | 101 …argspec: "args=[\'self\', \'outer_axis\', \'inner_axis\'], varargs=None, keywords=None, defaults=N…
|
| /external/tensorflow/tensorflow/tools/api/golden/v1/ |
| D | tensorflow.-ragged-tensor.pbtxt | 101 …argspec: "args=[\'self\', \'outer_axis\', \'inner_axis\'], varargs=None, keywords=None, defaults=N…
|