Searched defs:subshapes (Results 1 – 3 of 3) sorted by relevance
275 std::vector<xla::Shape> subshapes; in ExtractXlaShape() local1171 std::vector<xla::Shape> subshapes; in ExportXlaOp() local1226 std::vector<xla::Shape> subshapes; in ExportXlaOp() local1284 std::vector<xla::Shape> subshapes; in ExportXlaOp() local1831 auto subshapes = shape->mutable_tuple_shapes(); in ConvertLayout() local1924 auto subshapes = shape->mutable_tuple_shapes(); in ConvertInfeedtLayout() local
46 std::vector<const Shape*> subshapes; in FindMaxShape() local
97 std::vector<Shape> subshapes; in MakePartitionedShape() local119 std::vector<Shape> subshapes; in MakeNonPaddedShapeForGivenPartition() local