Searched refs:as_dimension (Results 1 – 6 of 6) sorted by relevance
202 other = as_dimension(other)212 other = as_dimension(other)248 other = as_dimension(other)293 other = as_dimension(other)318 other = as_dimension(other)353 other = as_dimension(other)368 other = as_dimension(other)393 other = as_dimension(other)432 other = as_dimension(other)449 other = as_dimension(other)[all …]
128 tensor_shape.as_dimension(tensor_shape.Dimension(12)))129 self.assertEqual(tensor_shape.Dimension(12), tensor_shape.as_dimension(12))132 tensor_shape.as_dimension(tensor_shape.Dimension(None)).value)134 tensor_shape.as_dimension(None).value)
120 input_size = tuple([tensor_shape.as_dimension(x).value for x in input_size])121 filter_size = tuple([tensor_shape.as_dimension(x).value for x in filter_size])
51 self._shard_dimension = tensor_shape.as_dimension(120 self._shard_dimension = tensor_shape.as_dimension(shard_dimension)
1516 return tensor_shape.as_dimension(batch_input_shape[0]).value
1785 ds_batch_size = tensor_shape.as_dimension(