Searched defs:tuple_shapes (Results 1 – 11 of 11) sorted by relevance
131 tuple_shapes=None, argument264 def tuple_shapes(self): member in InfeedQueue268 def set_tuple_shapes(self, tuple_shapes): argument776 tuple_shapes=None, argument
44 std::vector<Shape> tuple_shapes) in Shape()145 const Shape& tuple_shapes(int index) const { return tuple_shapes_.at(index); } in tuple_shapes() function152 const std::vector<Shape>& tuple_shapes() const { return tuple_shapes_; } in tuple_shapes() function
219 repeated ShapeProto tuple_shapes = 4; field
645 const auto& tuple_shapes = shape.tuple_shapes(); in HumanString() local677 const auto& tuple_shapes = shape.tuple_shapes(); in HumanStringWithLayout() local
75 def tuple_shapes(self): member in Shape
55 const auto& tuple_shapes = shape.tuple_shapes(); in BuildFakeDataOpOnDevice() local
244 absl::InlinedVector<Shape, 2> tuple_shapes; in RewriteBatchDimensionLargerThanTile() local
293 std::vector<Shape> tuple_shapes = { in InvertDiagonalBlocks() local
2283 const auto& tuple_shapes = to_apply.result().tuple_shapes(); in InferReduceShape() local
259 std::vector<xla::Shape> tuple_shapes; in FromC() local
249 XLA_Shape* tuple_shapes; // owned member