Searched refs:_set_shape (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_math_ops.py | 995 result._set_shape(a.shape[:-2] + a.shape[-1:] + [None])
997 result._set_shape(a.shape[:-2] + a.shape[-2:-1] + [None])
999 result._set_shape(b.shape[:-2] + [None] + b.shape[-2:-1])
1001 result._set_shape(b.shape[:-2] + [None] + b.shape[-1:])
|
| D | ragged_tensor_test.py | 1961 rt1._set_shape(shape)
1985 rt2._set_shape(shape)
1997 rt1._set_shape([2, 3, 1])
2003 rt2._set_shape([2, 3, 1])
2010 rt._set_shape([None, None, 5])
2012 rt._set_shape([None, 5, None])
2014 rt._set_shape([5, None, None])
|
| D | ragged_tensor.py | 1512 def _set_shape(self, shape): member in RaggedTensor
2520 result._set_shape(self._shape) # pylint: disable=protected-access
2594 result._set_shape(self._shape) # pylint: disable=protected-access
|