Inheritance diagram for torch::jit::fuser::cuda::FusedKernelCUDA:

Public Member Functions
	FusedKernelCUDA (int16_t device, 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)

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

at::Backend	backend () const override

Public Member Functions inherited from torch::jit::fuser::FusedKernel
	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)

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

Additional Inherited Members
Protected Attributes inherited from torch::jit::fuser::FusedKernel
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 22 of file fused_kernel.h.

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

Public Member Functions