Method MatchesLayoutInShape

tensorflow/compiler/xla/shape_layout.cc:49–56

}

bool ShapeLayout::MatchesLayoutInShape(const Shape& shape,
                                       bool minor_to_major_only) const {
  auto equal = Shape::Equal().IgnoreDynamicDimension();
  if (minor_to_major_only) {
    equal.MinorToMajorOnlyInLayout();
  }
  return equal(shape, shape_);
}

const Layout& ShapeLayout::layout() const {
  CHECK(LayoutIsSet());