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

Public Member Functions

 PartitionDesc (const TensorDesc &_desc, size_t _nSubTensors, size_t _dim)
 
bool isNoop () const
 
size_t nSubTensors () const
 
size_t dim () const
 
std::shared_ptr< TensorDescsubTensorDesc ()
 
const std::shared_ptr< TensorDescsubTensorDesc () const
 

Detailed Description

Definition at line 19 of file partition_desc.h.


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