Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dir_emitter_unnested.cc1601 auto has_fortran_layout = [](mlir::DenseIntElementsAttr layout_attr) { in EmitTriangularSolveFromMlir() argument
1602 int64_t n = layout_attr.getNumElements(); in EmitTriangularSolveFromMlir()
1603 return layout_attr.getValue<int64_t>({0}) == n - 2 && in EmitTriangularSolveFromMlir()
1604 layout_attr.getValue<int64_t>({1}) == n - 1; in EmitTriangularSolveFromMlir()