Searched refs:MinorToMajor (Results 1 – 19 of 19) sorted by relevance
78 for (auto dimension : LayoutUtil::MinorToMajor(shape)) { in MultidimensionalIndexToLinearIndex()108 for (auto dimension : LayoutUtil::MinorToMajor(shape)) { in LinearIndexToMultidimensionalIndex()134 for (auto dim : LayoutUtil::MinorToMajor(shape)) { in GetDimensionStride()
120 static absl::Span<const int64> MinorToMajor(const Shape& shape);121 static absl::Span<const int64> MinorToMajor(const Layout& layout);
314 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil320 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil
1520 absl::Span<const int64> minor_to_major_a = LayoutUtil::MinorToMajor(a); in FindTranspose021()1523 absl::Span<const int64> minor_to_major_b = LayoutUtil::MinorToMajor(b); in FindTranspose021()
776 const int64 rank = LayoutUtil::MinorToMajor(shape).size();
697 for (auto index : LayoutUtil::MinorToMajor(shape())) { in Transpose()738 LayoutUtil::MinorToMajor(shape())); in Slice()
45 for (int i = LayoutUtil::MinorToMajor(shape_).size() - 1; i >= 0; --i) { in EmitIndexAndSetExitBasicBlock()
1928 for (int i = LayoutUtil::MinorToMajor(reduce->shape()).size() - 1; i > 0; in EmitVectorizedReduce()1976 CHECK_GT(LayoutUtil::MinorToMajor(reduce->shape()).size(), 1); in EmitVectorizedReduce()1979 CHECK_EQ(LayoutUtil::MinorToMajor(reduce->shape()).size(), 1); in EmitVectorizedReduce()2197 for (int64 dim : LayoutUtil::MinorToMajor(layout)) { in HandleSlice()2650 auto output_min2maj = LayoutUtil::MinorToMajor(output_layout); in EmitFastConcatenate()
102 for (int i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitIndexAndSetExitBasicBlock()
238 for (auto dimension : LayoutUtil::MinorToMajor(shape)) { in SourceIndexOfBitcast()373 for (int64 i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitArrayElementAddress()
266 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in EmitOperandArrayLoopNest()
315 for (int64 dimension : LayoutUtil::MinorToMajor(keys_shape)) { in EmitSortInPlace()
230 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in ShapeToIrType()
38 for (auto dim : xla::LayoutUtil::MinorToMajor(shape)) { in PopulateInfeedLayoutVector()
80 for (int64 cur_dim : LayoutUtil::MinorToMajor(shape)) { in PartitionShapeByMiddleDimensions()
1078 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()1139 LayoutUtil::MinorToMajor(operand_layout)); in ChooseOutputLayoutFromOperandLayout()1489 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
4292 (PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()4294 PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
136 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F()
2685 for (auto dim : LayoutUtil::MinorToMajor(shape)) {