Inheritance diagram for torch::jit::fuser::FusedKernel:

Public Member Functions
	TH_DISALLOW_COPY_AND_ASSIGN (FusedKernel)

	FusedKernel (std::string name, std::string code, std::vector< TensorDesc > input_desc, std::vector< TensorDesc > output_desc, std::vector< PartitionDesc > chunk_desc, std::vector< PartitionDesc > concat_desc, bool has_random)

virtual void	launch_raw (const uint32_t numel, std::vector< void * > &arguments) const =0

virtual at::Backend	backend () const =0

const std::string &	name () const

const std::string &	code () const

const std::vector< TensorDesc > &	inputDesc () const

const std::vector< TensorDesc > &	outputDesc () const

const std::vector< PartitionDesc > &	chunkDesc () const

const std::vector< PartitionDesc > &	concatDesc () const

bool	hasRandom () const

Protected Attributes
const std::string	name_

const std::string	code_

const std::vector< TensorDesc >	input_desc_

const std::vector< TensorDesc >	output_desc_

const std::vector< PartitionDesc >	chunk_desc_

const std::vector< PartitionDesc >	concat_desc_

const bool	has_random_

Detailed Description

Definition at line 16 of file fused_kernel.h.

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

Public Member Functions