Searched refs:outer_axis (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_merge_dims_op_test.py | 184 outer_axis, argument
192 actual = x.merge_dims(outer_axis, inner_axis)
194 if outer_axis >= 0 and inner_axis >= 0:
196 x.shape.rank - (inner_axis - outer_axis))
199 if outer_axis >= 0 and inner_axis >= 0:
200 actual_with_neg_axis = x.merge_dims(outer_axis - x.shape.rank,
205 if (not context.executing_eagerly() and outer_axis >= 0 and
212 outer_axis, inner_axis)
261 outer_axis, argument
268 self.evaluate(x.merge_dims(outer_axis, inner_axis))
|
| D | ragged_tensor.py | 1388 def merge_dims(self, outer_axis, inner_axis): argument
1421 outer_axis = array_ops.get_positive_axis(
1422 outer_axis,
1431 if not outer_axis <= inner_axis:
1433 "inner_axis (%d)" % (outer_axis, inner_axis))
1434 return merge_dims(self, outer_axis, inner_axis)
2715 def merge_dims(value, outer_axis, inner_axis): argument
2729 if outer_axis == inner_axis:
2733 while outer_axis == 0 and isinstance(value, RaggedTensor):
2743 new_shape = old_shape[:outer_axis] + [-1] + old_shape[inner_axis + 1:]
[all …]
|
| /external/tensorflow/tensorflow/python/ops/structured/ |
| D | structured_tensor.py | 980 def merge_dims(self, outer_axis, inner_axis): argument
1006 outer_axis = array_ops.get_positive_axis(
1007 outer_axis,
1016 if not outer_axis <= inner_axis:
1018 'inner_axis (%d)' % (outer_axis, inner_axis))
1019 return _merge_dims(self, outer_axis, inner_axis)
1509 def _merge_dims(value, outer_axis, inner_axis): argument
1511 assert outer_axis < inner_axis
1513 return ragged_tensor.merge_dims(value, outer_axis, inner_axis)
1518 fields = dict((k, _merge_dims(v, outer_axis, inner_axis))
[all …]
|
| D | structured_tensor_test.py | 898 outer_axis=0, inner_axis=1,
903 outer_axis=0, inner_axis=1,
908 outer_axis=1, inner_axis=2,
913 outer_axis=0, inner_axis=2,
919 outer_axis=0, inner_axis=1,
926 outer_axis=0, inner_axis=2,
933 outer_axis=0, inner_axis=3,
940 outer_axis=1, inner_axis=2,
947 outer_axis=1, inner_axis=3,
954 outer_axis=2, inner_axis=3,
[all …]
|
| /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…
|
| /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…
|