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:

• torch/csrc/jit/passes/to_batch.h
• torch/csrc/jit/passes/to_batch.cpp