Searched defs:tuple_shapes (Results 1 – 4 of 4) sorted by relevance
45 tuple_shapes=None, argument166 def tuple_shapes(self): member in InfeedQueue170 def set_tuple_shapes(self, tuple_shapes): argument
422 std::vector<xla::Shape> tuple_shapes = { in TriangularSolveLeftLooking() local
223 def tuple_shapes(self): member in Shape
188 repeated Shape tuple_shapes = 4; field