HomeSort by relevance Sort by last modified time
    Searched defs:tuple_shapes (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/tpu/python/tpu/
tpu_feed.py 45 tuple_shapes=None,
58 tuple_shapes: if not None, a list of shapes of the elements of the queue.
66 number_of_tuple_arguments, tuple_types, tuple_shapes, and
68 tuple_shapes, or shard_dimensions is not equal to
71 TypeError: if any element of tuple_types or tuple_shapes can't
81 elif tuple_shapes is not None:
82 number_of_tuple_elements = len(tuple_shapes)
102 if tuple_shapes is not None:
103 self.set_tuple_shapes(tuple_shapes)
116 if self.tuple_shapes is not None
166 def tuple_shapes(self): member in class:InfeedQueue
    [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/lib/
triangular_solve.cc 422 std::vector<xla::Shape> tuple_shapes = { local
431 xla::Shape tuple_shape = xla::ShapeUtil::MakeTupleShape(tuple_shapes);
  /external/tensorflow/tensorflow/compiler/xla/python/
xla_client.py 223 def tuple_shapes(self): member in class:Shape
242 children = tuple(child.map_leaves(f) for child in self.tuple_shapes())
    [all...]

Completed in 649 milliseconds