Function Coefficient

tensorflow/compiler/xla/client/lib/tridiagonal.cc:117–119

}

XlaOp Coefficient(XlaOp operand, int64 i) {
  return SliceInMinorDims(operand, /*start=*/{i}, /*end=*/{i + 1});
}

}  // namespace