Home
last modified time | relevance | path

Searched refs:mutable_minor_to_major (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/
Dshape.cc107 layout_.mutable_minor_to_major()->erase( in DeleteDimension()
108 layout_.mutable_minor_to_major()->begin() + i); in DeleteDimension()
112 (*layout_.mutable_minor_to_major())[i] -= 1; in DeleteDimension()
Dlayout.cc66 proto.mutable_minor_to_major()->Reserve(minor_to_major_size()); in ToProto()
Dlayout_util.cc107 std::vector<int64>* minor_to_major = layout.mutable_minor_to_major(); in CreateDefaultLayoutForRank()
151 auto* minor_to_major = shape->mutable_layout()->mutable_minor_to_major(); in SetToDefaultLayout()
Dlayout.h168 std::vector<int64>* mutable_minor_to_major() { return &minor_to_major_; } in mutable_minor_to_major() function
Dshape_util.cc1297 *output_shape_with_layout.mutable_layout()->mutable_minor_to_major() = in AlignLayouts()
Dliteral_test.cc1452 shape_proto->mutable_layout()->mutable_minor_to_major(); in SetDefaultLayoutOnProto()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dcusolver_rewriter.cc46 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0), in SetFortranLayout()
47 shape->mutable_layout()->mutable_minor_to_major()->at(1)); in SetFortranLayout()
Dgpu_layout_assignment.cc254 std::swap(shape->mutable_layout()->mutable_minor_to_major()->at(0), in AddBackendConstraints()
255 shape->mutable_layout()->mutable_minor_to_major()->at(1)); in AddBackendConstraints()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dcopy_test.cc137 *literal_layout->mutable_minor_to_major() = { in XLA_TEST_F()