HomeSort by relevance Sort by last modified time
    Searched full:as_dimension (Results 1 - 4 of 4) sorted by null

  /external/tensorflow/tensorflow/python/framework/
tensor_shape.py 51 other = as_dimension(other)
61 other = as_dimension(other)
97 other = as_dimension(other)
139 other = as_dimension(other)
164 other = as_dimension(other)
188 other = as_dimension(other)
212 other = as_dimension(other)
236 other = as_dimension(other)
276 other = as_dimension(other)
301 other = as_dimension(other
383 def as_dimension(value): function
    [all...]
tensor_shape_test.py 119 tensor_shape.as_dimension(tensor_shape.Dimension(12)))
120 self.assertEqual(tensor_shape.Dimension(12), tensor_shape.as_dimension(12))
123 tensor_shape.as_dimension(tensor_shape.Dimension(None)).value)
125 tensor_shape.as_dimension(None).value)
common_shapes.py 108 input_size = tuple([tensor_shape.as_dimension(x).value for x in input_size])
109 filter_size = tuple([tensor_shape.as_dimension(x).value for x in filter_size])
  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_sharding.py 51 self._shard_dimension = tensor_shape.as_dimension(
120 self._shard_dimension = tensor_shape.as_dimension(shard_dimension)

Completed in 198 milliseconds