Cholesky.hpp File Reference

Factorise a positive definite matrix using Cholesky decomposition. More...

#include "poplin/MatMul.hpp"
#include <poplar/Graph.hpp>
#include <poplar/Program.hpp>

Go to the source code of this file.

Namespaces
namespace	poplin
	Linear algebra functions.

Functions
std::vector< std::pair< MatMulParams, poplar::OptionFlags > >	poplin::getCholeskyMatMulPrePlanParameters (const poplar::Type &type, const std::vector< std::size_t > &shape, bool lower, poplar::OptionFlags options)
	Plan matrix multiplication for the Cholesky factoriser. More...
poplar::Tensor	poplin::createCholeskyInput (poplar::Graph &graph, const poplar::Type &type, const std::vector< std::size_t > &shape, bool lower, const poplar::DebugContext &debugContext, const poplar::OptionFlags &options={}, PlanningCache *cache=nullptr)
	Create a tensor that is used as the input for the Cholesky factoriser. More...
poplar::Tensor	poplin::cholesky (poplar::Graph &graph, const poplar::Tensor &a, bool lower, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, poplar::OptionFlags options={}, PlanningCache *cache=nullptr)
	Computes Cholesky factor for a symmetric positive definite matrix. More...
void	poplin::choleskyInPlace (poplar::Graph &graph, const poplar::Tensor &a, bool lower, poplar::program::Sequence &prog, const poplar::DebugContext &debugContext={}, poplar::OptionFlags options={}, PlanningCache *cache=nullptr)
	Computes Cholesky factor in place for a symmetric positive definite matrix. More...

Detailed Description

Factorise a positive definite matrix using Cholesky decomposition.