Searched refs:minor_to_major_size (Results 1 – 13 of 13) sorted by relevance
214 if (layout.minor_to_major_size() != 0) { in ValidateLayoutForShape()228 if (layout.minor_to_major_size() != shape.rank()) { in ValidateLayoutForShape()232 layout.minor_to_major_size(), shape.rank(), in ValidateLayoutForShape()335 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Major()337 layout.minor_to_major_size() - 1 - physical_dimension_number); in Major()344 CHECK_LT(physical_dimension_number, layout.minor_to_major_size()); in Minor()350 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()
172 int minor_to_major_size() const { return minor_to_major_.size(); } in minor_to_major_size() function
58 const int minor_to_major_size = minor_to_major.size(); in MakeLayout() local59 if (dim < 0 || dim >= minor_to_major_size) { in MakeLayout()213 const int64 minor_to_major_size = minor_to_major.size(); in GetShapeWithLayout() local214 if (rank != minor_to_major_size) { in GetShapeWithLayout()
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()
50 for (int64 i = 0; i < layout.minor_to_major_size(); ++i) { in Delinearize()65 if (i < layout.minor_to_major_size() - 1) { in Delinearize()83 for (int64 i = 0; i < layout.minor_to_major_size(); ++i) { in Delinearize()91 if (i < layout.minor_to_major_size() - 1) { in Delinearize()
43 int n = shape->mutable_layout()->minor_to_major_size(); in SetFortranLayout()
285 int n = shape->mutable_layout()->minor_to_major_size(); in AddBackendConstraints()
135 ASSERT_EQ(2, literal_layout->minor_to_major_size()); in XLA_TEST_F()
734 std::vector<int64_t> minor_to_major(layout.minor_to_major_size()); in EmitDnnConvolution()
4623 if (layout.minor_to_major_size() != result->rank()) { in ParseShape()4627 result->rank(), layout.minor_to_major_size())); in ParseShape()
967 CHECK_EQ(num_dimensions, layout.minor_to_major_size()); in ComputeStrides()