Searched refs:subshapes (Results 1 – 4 of 4) sorted by relevance
774 subshapes given loop bounds. This is used to answer the question: "given775 an iteration space over the codomain, what are the subshapes of the779 compute subshapes even in cases where the concatenation of indexing maps799 hyperrectangular loop iteration space given all the operand subshapes.
150 std::vector<Shape> subshapes; in MakePartitionedShape() local152 subshapes.push_back( in MakePartitionedShape()156 return ShapeUtil::MakeTupleShape(subshapes); in MakePartitionedShape()170 std::vector<Shape> subshapes; in MakeNonPaddedShapeForGivenPartition() local172 subshapes.push_back(MakeNonPaddedShapeForGivenPartition( in MakeNonPaddedShapeForGivenPartition()176 return ShapeUtil::MakeTupleShape(subshapes); in MakeNonPaddedShapeForGivenPartition()
1166 auto subshapes = shape->mutable_tuple_shapes(); in ConvertLayout() local1167 if (layout.size() != subshapes->size()) { in ConvertLayout()1169 << ", but found " << subshapes->size(); in ConvertLayout()1172 for (int i = 0; i < subshapes->size(); i++) { in ConvertLayout()1183 if (failed(ConvertLayout(op, c, subshapes->Mutable(i)))) { in ConvertLayout()
80 std::vector<xla::Shape> subshapes(tuple_node.tuples_size(), dummy); in ParseTupleNode()82 xla::ShapeUtil::MakeTupleShape(subshapes); in ParseTupleNode()