Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions
torch::jit::fuser::KernelSpec Struct Reference

Public Member Functions

 KernelSpec (const int64_t _key, const std::shared_ptr< Graph > &_graph)
 
int64_t key () const
 
std::shared_ptr< Graphgraph () const
 
const Codecode () 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: