TMVA::Experimental::SOFIE::UTILITY Namespace Reference

Functions
bool	AreSameShape (const std::vector< Dim > &, const std::vector< Dim > &)
bool	AreSameShape (const std::vector< size_t > &, const std::vector< Dim > &)
bool	AreSameShape (const std::vector< size_t > &, const std::vector< size_t > &)
template<typename T >
T *	BroadcastConvBias (const T *data, const size_t channel, const std::vector< size_t > &targetShape)
template<typename T , class ConstContT = std::span<const T>>
void	BroadcastTensor (ConstContT data, const std::vector< size_t > &shape, const std::vector< size_t > &targetShape, T *broadcastedData)
std::string	Clean_name (std::string input_tensor_name)
std::vector< Dim >	ComputeStrideFromShape (const std::vector< Dim > &shape)
std::vector< size_t >	ComputeStrideFromShape (const std::vector< size_t > &shape)
	compute stride of a tensor given its shape (assume layout is row-major)
template<typename T >
T *	CreateBroadcastTensor (const T *data, const std::vector< size_t > &shape, const std::vector< size_t > &targetShape, size_t targetLength)
std::pair< int, std::vector< Dim > >	MultidirectionalBroadcastShape (std::vector< Dim > &, std::vector< Dim > &)
std::pair< int, std::vector< size_t > >	MultidirectionalBroadcastShape (std::vector< size_t > &, std::vector< size_t > &)
std::vector< size_t >	MultidirectionalBroadcastShape (std::vector< std::vector< size_t > >)
template<typename T >
T *	UnidirectionalBroadcast (const T *data, const std::vector< size_t > &shape, const std::vector< size_t > &targetShape)
template<typename T >
void	UnidirectionalBroadcast (const T *data, const std::vector< size_t > &shape, const std::vector< size_t > &targetShape, T *broadcastedData)
std::vector< size_t >	UnidirectionalBroadcastShape (std::vector< size_t > &, std::vector< size_t > &)

Function Documentation

AreSameShape() [1/3]

bool TMVA::Experimental::SOFIE::UTILITY::AreSameShape	(	const std::vector< Dim > &	shapeA,
		const std::vector< Dim > &	shapeB )

Definition at line 225 of file SOFIE_common.cxx.

AreSameShape() [2/3]

bool TMVA::Experimental::SOFIE::UTILITY::AreSameShape	(	const std::vector< size_t > &	shapeA,
		const std::vector< Dim > &	shapeB )

Definition at line 213 of file SOFIE_common.cxx.

AreSameShape() [3/3]

bool TMVA::Experimental::SOFIE::UTILITY::AreSameShape	(	const std::vector< size_t > &	shapeA,
		const std::vector< size_t > &	shapeB )

Definition at line 202 of file SOFIE_common.cxx.

BroadcastConvBias()

template<typename T >

T * TMVA::Experimental::SOFIE::UTILITY::BroadcastConvBias	(	const T *	data,
		const size_t	channel,
		const std::vector< size_t > &	targetShape )

Definition at line 368 of file SOFIE_common.hxx.

BroadcastTensor()

template<typename T , class ConstContT = std::span<const T>>

void TMVA::Experimental::SOFIE::UTILITY::BroadcastTensor	(	ConstContT	data,
		const std::vector< size_t > &	shape,
		const std::vector< size_t > &	targetShape,
		T *	broadcastedData )

Definition at line 410 of file SOFIE_common.hxx.

Clean_name()

std::string TMVA::Experimental::SOFIE::UTILITY::Clean_name

(

std::string

input_tensor_name

)

Definition at line 512 of file SOFIE_common.cxx.

ComputeStrideFromShape() [1/2]

std::vector< Dim > TMVA::Experimental::SOFIE::UTILITY::ComputeStrideFromShape

(

const std::vector< Dim > &

shape

)

Definition at line 530 of file SOFIE_common.cxx.

ComputeStrideFromShape() [2/2]

std::vector< size_t > TMVA::Experimental::SOFIE::UTILITY::ComputeStrideFromShape

(

const std::vector< size_t > &

shape

)

compute stride of a tensor given its shape (assume layout is row-major)

Definition at line 520 of file SOFIE_common.cxx.

CreateBroadcastTensor()

template<typename T >

T * TMVA::Experimental::SOFIE::UTILITY::CreateBroadcastTensor	(	const T *	data,
		const std::vector< size_t > &	shape,
		const std::vector< size_t > &	targetShape,
		size_t	targetLength )

Definition at line 473 of file SOFIE_common.hxx.

MultidirectionalBroadcastShape() [1/3]

std::pair< int, std::vector< Dim > > TMVA::Experimental::SOFIE::UTILITY::MultidirectionalBroadcastShape	(	std::vector< Dim > &	shapeA,
		std::vector< Dim > &	shapeB )

Definition at line 434 of file SOFIE_common.cxx.

MultidirectionalBroadcastShape() [2/3]

std::pair< int, std::vector< size_t > > TMVA::Experimental::SOFIE::UTILITY::MultidirectionalBroadcastShape	(	std::vector< size_t > &	shapeA,
		std::vector< size_t > &	shapeB )

Definition at line 369 of file SOFIE_common.cxx.

MultidirectionalBroadcastShape() [3/3]

std::vector< size_t > TMVA::Experimental::SOFIE::UTILITY::MultidirectionalBroadcastShape

(

std::vector< std::vector< size_t > >

shape

)

Definition at line 237 of file SOFIE_common.cxx.

UnidirectionalBroadcast() [1/2]

template<typename T >

T * TMVA::Experimental::SOFIE::UTILITY::UnidirectionalBroadcast	(	const T *	data,
		const std::vector< size_t > &	shape,
		const std::vector< size_t > &	targetShape )

Definition at line 483 of file SOFIE_common.hxx.

UnidirectionalBroadcast() [2/2]

template<typename T >

void TMVA::Experimental::SOFIE::UTILITY::UnidirectionalBroadcast	(	const T *	data,
		const std::vector< size_t > &	shape,
		const std::vector< size_t > &	targetShape,
		T *	broadcastedData )

Definition at line 497 of file SOFIE_common.hxx.

UnidirectionalBroadcastShape()

std::vector< size_t > TMVA::Experimental::SOFIE::UTILITY::UnidirectionalBroadcastShape	(	std::vector< size_t > &	shapeA,
		std::vector< size_t > &	shapeB )

Definition at line 417 of file SOFIE_common.cxx.