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()132 for (auto dim : LayoutUtil::MinorToMajor(shape)) { in GetDimensionStride()
134 static absl::Span<const int64> MinorToMajor(const Shape& shape);135 static absl::Span<const int64> MinorToMajor(const Layout& layout);
331 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil337 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil
734 const int64 rank = LayoutUtil::MinorToMajor(shape).size();
703 for (auto index : LayoutUtil::MinorToMajor(shape())) { in Transpose()744 LayoutUtil::MinorToMajor(shape())); in Slice()
87 absl::Span<const int64> minor_to_major_a = LayoutUtil::MinorToMajor(a); in FindTranspose021()90 absl::Span<const int64> minor_to_major_b = LayoutUtil::MinorToMajor(b); in FindTranspose021()
102 for (int i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitIndexAndSetExitBasicBlock()
233 for (auto dimension : LayoutUtil::MinorToMajor(shape)) { in SourceIndexOfBitcast()371 for (int64 i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitArrayElementAddress()
264 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in EmitOperandArrayLoopNest()
313 for (int64 dimension : LayoutUtil::MinorToMajor(keys_shape)) { in EmitSortInPlace()
229 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in ShapeToIrType()
45 for (int i = LayoutUtil::MinorToMajor(shape_).size() - 1; i >= 0; --i) { in EmitIndexAndSetExitBasicBlock()
1725 for (int i = LayoutUtil::MinorToMajor(reduce->shape()).size() - 1; i > 0; in EmitVectorizedReduce()1773 CHECK_GT(LayoutUtil::MinorToMajor(reduce->shape()).size(), 1); in EmitVectorizedReduce()1776 CHECK_EQ(LayoutUtil::MinorToMajor(reduce->shape()).size(), 1); in EmitVectorizedReduce()1954 for (int64 dim : LayoutUtil::MinorToMajor(layout)) { in HandleSlice()2407 auto output_min2maj = LayoutUtil::MinorToMajor(output_layout); in EmitFastConcatenate()
38 for (auto dim : xla::LayoutUtil::MinorToMajor(shape)) { in PopulateInfeedLayoutVector()
1021 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()1082 LayoutUtil::MinorToMajor(operand_layout)); in ChooseOutputLayoutFromOperandLayout()1400 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
3587 (PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()3589 PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
136 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F()
2812 for (auto dim : LayoutUtil::MinorToMajor(shape)) { in MakeDimMultipliers()
3508 return PositionInContainer(LayoutUtil::MinorToMajor(input_shape), dim_a) < in GetDimensionsToKeepMinorToMajor()3509 PositionInContainer(LayoutUtil::MinorToMajor(input_shape), dim_b); in GetDimensionsToKeepMinorToMajor()3537 LayoutUtil::MinorToMajor(input_shape); in GetReductionToVectorDimensions()