Searched defs:tuple_shapes (Results 1 – 8 of 8) sorted by relevance
136 tuple_shapes=None, argument268 def tuple_shapes(self): member in InfeedQueue272 def set_tuple_shapes(self, tuple_shapes): argument775 tuple_shapes=None, argument
44 std::vector<Shape> tuple_shapes) in Shape()135 const Shape& tuple_shapes(int index) const { return tuple_shapes_.at(index); } in tuple_shapes() function142 const std::vector<Shape>& tuple_shapes() const { return tuple_shapes_; } in tuple_shapes() function
214 repeated ShapeProto tuple_shapes = 4; field
611 const auto& tuple_shapes = shape.tuple_shapes(); in HumanString() local641 const auto& tuple_shapes = shape.tuple_shapes(); in HumanStringWithLayout() local
82 def tuple_shapes(self): member in Shape
238 std::vector<xla::Shape> tuple_shapes; in FromC() local
226 XLA_Shape* tuple_shapes; // owned member
300 std::vector<Shape> tuple_shapes = { in InvertDiagonalBlocks() local