Searched defs:tuple_shapes (Results 1 – 7 of 7) 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()129 const Shape& tuple_shapes(int index) const { return tuple_shapes_.at(index); } in tuple_shapes() function136 const std::vector<Shape>& tuple_shapes() const { return tuple_shapes_; } in tuple_shapes() function
214 repeated ShapeProto tuple_shapes = 4; field
82 def tuple_shapes(self): member in Shape
238 std::vector<xla::Shape> tuple_shapes; in FromC() local
223 XLA_Shape* tuple_shapes; // owned member
300 std::vector<Shape> tuple_shapes = { in InvertDiagonalBlocks() local