Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dlayout_assignment.h322 static bool IsAtMostRank1(const Shape& shape);
Dlayout_assignment.cc1491 if (IsAtMostRank1(instruction->shape())) { in PropagateBufferConstraintToOperands()
1505 if (IsAtMostRank1(operand->shape())) { in PropagateBufferConstraintToOperands()
2323 bool LayoutAssignment::IsAtMostRank1(const Shape& shape) { in IsAtMostRank1() function in xla::LayoutAssignment
2328 return IsAtMostRank1(subshape); in IsAtMostRank1()