Home
last modified time | relevance | path

Searched refs:MinorToMajor (Results 1 – 19 of 19) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dindex_util.cc78 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()
Dlayout_util.h134 static absl::Span<const int64> MinorToMajor(const Shape& shape);
135 static absl::Span<const int64> MinorToMajor(const Layout& layout);
Dlayout_util.cc331 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil
337 /* static */ absl::Span<const int64> LayoutUtil::MinorToMajor( in MinorToMajor() function in xla::LayoutUtil
Dshape_util.h734 const int64 rank = LayoutUtil::MinorToMajor(shape).size();
Dliteral.cc703 for (auto index : LayoutUtil::MinorToMajor(shape())) { in Transpose()
744 LayoutUtil::MinorToMajor(shape())); in Slice()
/external/tensorflow/tensorflow/compiler/xla/service/llvm_ir/
Dkernel_tiling.cc87 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()
Dloop_emitter.cc102 for (int i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitIndexAndSetExitBasicBlock()
Dir_array.cc233 for (auto dimension : LayoutUtil::MinorToMajor(shape)) { in SourceIndexOfBitcast()
371 for (int64 i = 0; i < LayoutUtil::MinorToMajor(shape_).size(); ++i) { in EmitArrayElementAddress()
Dllvm_loop.cc264 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in EmitOperandArrayLoopNest()
Dsort_util.cc313 for (int64 dimension : LayoutUtil::MinorToMajor(keys_shape)) { in EmitSortInPlace()
Dllvm_util.cc229 for (int64 dimension : LayoutUtil::MinorToMajor(shape)) { in ShapeToIrType()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dparallel_loop_emitter.cc45 for (int i = LayoutUtil::MinorToMajor(shape_).size() - 1; i >= 0; --i) { in EmitIndexAndSetExitBasicBlock()
Dir_emitter.cc1725 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()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dshape_util.cc38 for (auto dim : xla::LayoutUtil::MinorToMajor(shape)) { in PopulateInfeedLayoutVector()
/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.cc1021 LayoutUtil::MinorToMajor(output_layout)); in ChooseOperandLayoutFromOutputLayout()
1082 LayoutUtil::MinorToMajor(operand_layout)); in ChooseOutputLayoutFromOperandLayout()
1400 LayoutUtil::MinorToMajor(buffer_constraint.layout()).size()) { in PropagateBufferConstraintToOperands()
Dalgebraic_simplifier.cc3587 (PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
3589 PositionInContainer(LayoutUtil::MinorToMajor(filter_shape), in SimplifyConvToDot()
Dcopy_insertion_test.cc136 auto minor_to_major = LayoutUtil::MinorToMajor(constant->shape()); in TEST_F()
Dhlo_evaluator_typed_visitor.h2812 for (auto dim : LayoutUtil::MinorToMajor(shape)) { in MakeDimMultipliers()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc3508 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()