Public Member Functions
	KernelSpec (const int64_t _key, const std::shared_ptr< Graph > &_graph)

int64_t	key () const

std::shared_ptr< Graph >	graph () const

const Code &	code () const

int64_t	nInputs () const

int64_t	nTensorInputs () const

std::vector< std::vector< int64_t > > &	inputBroadcastGroups ()

const std::vector< std::vector< int64_t > > &	inputBroadcastGroups () const

std::vector< PartitionInfo > &	inputChunks ()

const std::vector< PartitionInfo > &	inputChunks () const

std::vector< OutputMapAndSize > &	outputMapAndSizes ()

bool	hasRandom () const

c10::optional< std::shared_ptr< FusedKernel > >	findKernel (const ArgSpec &arg_spec) const

void	cacheKernel (const ArgSpec &arg_spec, std::shared_ptr< FusedKernel > kernel) const

Detailed Description

Definition at line 81 of file kernel_spec.h.

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

Public Member Functions