Public Member Functions
DSizes< DenseIndex, 3 >	dimensions (const Tensor< float, 3 > &input1, const Tensor< float, 3 > &input2) const

DSizes< DenseIndex, 3 >	dimensions (const TensorType &input1, const TensorType &input2) const

template<typename Output , typename Device >
void	eval (const Tensor< float, 3 > &input1, const Tensor< float, 3 > &input2, Output &output, const Device &device) const

template<typename Output , typename Device >
void	eval (const TensorType &input1, const TensorType &input2, Output &output, const Device &device) const

Detailed Description

Definition at line 61 of file cxx11_tensor_custom_op.cpp.

Member Function Documentation

template<typename TensorType >

DSizes<DenseIndex, 3> BatchMatMul< TensorType >::dimensions	(	const Tensor< float, 3 > &	input1,
		const Tensor< float, 3 > &	input2
	)		const

inline

Definition at line 62 of file cxx11_tensor_custom_op.cpp.

template<typename TensorType >

DSizes<DenseIndex, 3> BatchMatMul< TensorType >::dimensions	(	const TensorType &	input1,
		const TensorType &	input2
	)		const

inline

template<typename TensorType >

template<typename Output , typename Device >

void BatchMatMul< TensorType >::eval	(	const Tensor< float, 3 > &	input1,
		const Tensor< float, 3 > &	input2,
		Output &	output,
		const Device &	device
	)		const

inline

Definition at line 71 of file cxx11_tensor_custom_op.cpp.

template<typename TensorType >

template<typename Output , typename Device >

void BatchMatMul< TensorType >::eval	(	const TensorType &	input1,
		const TensorType &	input2,
		Output &	output,
		const Device &	device
	)		const

inline

The documentation for this struct was generated from the following files: