Searched refs:pivoting_ (Results 1 – 2 of 2) sorted by relevance
102 OP_REQUIRES_OK(context, context->GetAttr("partial_pivoting", &pivoting_)); in TridiagonalSolveOpGpuLinalg()203 auto buffer_function = pivoting_ in SolveWithGtsv()216 auto solver_function = pivoting_ ? &GpuSparse::Gtsv2<Scalar> in SolveWithGtsv()242 bool pivoting_; member in tensorflow::TridiagonalSolveOpGpuLinalg250 OP_REQUIRES_OK(context, context->GetAttr("partial_pivoting", &pivoting_)); in TridiagonalSolveOpGpu()267 pivoting_ // GtsvBatched doesn't do pivoting in Compute()360 bool pivoting_; member in tensorflow::TridiagonalSolveOpGpu
45 OP_REQUIRES_OK(context, context->GetAttr("partial_pivoting", &pivoting_)); in TridiagonalSolveOp()86 if (pivoting_) { in GetCostPerUnit()127 if (pivoting_) { in ComputeMatrix()223 bool pivoting_; member in tensorflow::TridiagonalSolveOp