Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Attributes
torch::jit::ToBatch Class Reference

Public Member Functions

TORCH_API void toBatch (Block *block, Block *res_block)

Static Public Attributes

static std::unordered_map< std::string, std::vector< std::shared_ptr< Graph > > > batch_operator_table

Detailed Description

Definition at line 11 of file to_batch.h.

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