Searched defs:device_shape (Results 1 – 1 of 1) sorted by relevance
342 xla::Shape* device_shape) { in ExpandTreeOfTuples()389 xla::Shape device_shape; in MakeTuple() local