Searched refs:minor_to_major_size (Results 1 – 11 of 11) sorted by relevance
212 if (layout.minor_to_major_size() != 0) { in ValidateLayoutForShape()226 if (layout.minor_to_major_size() != shape.rank()) { in ValidateLayoutForShape()230 layout.minor_to_major_size(), shape.rank(), in ValidateLayoutForShape()346 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Major()348 layout.minor_to_major_size() - 1 - physical_dimension_number); in Major()355 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Minor()361 std::vector<int64> logical_to_physical(layout.minor_to_major_size()); in MakeLogicalToPhysical()
51 layout.minor_to_major_.reserve(proto.minor_to_major_size()); in CreateFromProto()66 proto.mutable_minor_to_major()->Reserve(minor_to_major_size()); in ToProto()
153 int minor_to_major_size() const { return minor_to_major_.size(); } in minor_to_major_size() function
29 for (int i = shape_.layout().minor_to_major_size() - 1; i >= 0; --i) { in Run()109 shape_.layout().minor_to_major_size() - 1 - i); in ShapePartitionIterator()
174 shape.layout().minor_to_major_size() - 1 - i); in TEST_F()
44 int n = shape->mutable_layout()->minor_to_major_size(); in SetFortranLayout()
252 int n = shape->mutable_layout()->minor_to_major_size(); in AddBackendConstraints()
542 int n = layout.minor_to_major_size(); in HandleTriangularSolve()
45 for (int64 i = 0; i < layout.minor_to_major_size(); ++i) { in Delinearize()60 if (i < layout.minor_to_major_size() - 1) { in Delinearize()
135 ASSERT_EQ(2, literal_layout->minor_to_major_size()); in XLA_TEST_F()
3595 if (layout.minor_to_major_size() != result->rank()) { in ParseShape()3599 result->rank(), layout.minor_to_major_size())); in ParseShape()