Searched refs:as_dimension (Results 1 – 4 of 4) sorted by relevance
220 other = as_dimension(other)230 other = as_dimension(other)270 other = as_dimension(other)318 other = as_dimension(other)348 other = as_dimension(other)390 other = as_dimension(other)407 other = as_dimension(other)436 other = as_dimension(other)479 other = as_dimension(other)496 other = as_dimension(other)[all …]
126 tensor_shape.as_dimension(tensor_shape.Dimension(12)))127 self.assertEqual(tensor_shape.Dimension(12), tensor_shape.as_dimension(12))130 tensor_shape.as_dimension(tensor_shape.Dimension(None)).value)132 tensor_shape.as_dimension(None).value)
52 self._shard_dimension = tensor_shape.as_dimension(147 self._shard_dimension = tensor_shape.as_dimension(shard_dimension)
401 self._rank = tensor_shape.as_dimension(rank)