Caffe2 - C++ API
A deep learning, cross platform ML framework
Public Member Functions | Static Public Member Functions
caffe2::SmartTensorPrinter Class Reference

Public Member Functions

 SmartTensorPrinter (const std::string &tensor_name)
 
 SmartTensorPrinter (const std::string &tensor_name, const std::string &file_name)
 
 SmartTensorPrinter (const std::string &tensor_name, const std::string &file_name, int limit)
 
void Print (const Tensor< CPUContext > &tensor)
 
template<class Context >
void PrintMeta (const Tensor< Context > &tensor)
 
template<class Context >
void PrintTensorMeta (const Tensor< Context > &tensor)
 

Static Public Member Functions

static void PrintTensor (const Tensor< CPUContext > &tensor)
 

Detailed Description

Definition at line 27 of file smart_tensor_printer.h.


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