FM / utils

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/

/usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils

Name	Size	Mode	Actions
math/	-	0755	rm
threadpool/	-	0755	rm
bench_utils.h	854	0644	edit dl rm
cast.h	1110	0644	edit dl rm
cblas.h	34183	0644	edit dl rm
conversions.h	661	0644	edit dl rm
cpuid.h	2560	0644	edit dl rm
cpu_neon.h	1578	0644	edit dl rm
eigen_utils.h	6706	0644	edit dl rm
filler.h	3987	0644	edit dl rm
fixed_divisor.h	3634	0644	edit dl rm
knobs.h	532	0644	edit dl rm
knob_patcher.h	715	0644	edit dl rm
map_utils.h	635	0644	edit dl rm
math-detail.h	1916	0644	edit dl rm
math.h	15251	0644	edit dl rm
murmur_hash3.h	909	0644	edit dl rm
proto_convert.h	263	0644	edit dl rm
proto_utils.h	13254	0644	edit dl rm
proto_wrap.h	378	0644	edit dl rm
signal_handler.h	698	0644	edit dl rm
simple_queue.h	2602	0644	edit dl rm
smart_tensor_printer.h	1402	0644	edit dl rm
string_utils.h	1206	0644	edit dl rm
zmq_helper.h	3020	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/utils/smart_tensor_printer.h (1402B)

#pragma once

#include "caffe2/core/tensor.h"

namespace caffe2 {

// This is a wrapper around the TensorPrinter that doesn't require the user to
// explicit specify the type of the tensor while calling the Print() method.
// It also supports a convenience function with a default constructed printer as
// a static method.
class TORCH_API SmartTensorPrinter {
 public:
  // The proliferation of constructors is to give the feature parity with
  // TensorPrinter
  // yet not repeat the default arguments explicitly in case they change in the
  // future.
  SmartTensorPrinter() = default;

explicit 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& tensor);

void PrintMeta(const Tensor& tensor) {
    tensorPrinter_.PrintMeta(tensor);
  }

// Uses a default constructed SmartTensorPrinter
  static void PrintTensor(const Tensor& tensor);

// Uses a default constructed SmartTensorPrinter
  void PrintTensorMeta(const Tensor& tensor) {
    DefaultTensorPrinter().PrintMeta(tensor);
  }

private:
  // Returns a thread local default constructed TensorPrinter
  static SmartTensorPrinter& DefaultTensorPrinter();

TensorPrinter tensorPrinter_;
};
}

#pragma once #include "caffe2/core/tensor.h" namespace caffe2 { // This is a wrapper around the TensorPrinter that doesn't require the user to // explicit specify the type of the tensor while calling the Print() method. // It also supports a convenience function with a default constructed printer as // a static method. class TORCH_API SmartTensorPrinter { public: // The proliferation of constructors is to give the feature parity with // TensorPrinter // yet not repeat the default arguments explicitly in case they change in the // future. SmartTensorPrinter() = default; explicit 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& tensor); void PrintMeta(const Tensor& tensor) { tensorPrinter_.PrintMeta(tensor); } // Uses a default constructed SmartTensorPrinter static void PrintTensor(const Tensor& tensor); // Uses a default constructed SmartTensorPrinter void PrintTensorMeta(const Tensor& tensor) { DefaultTensorPrinter().PrintMeta(tensor); } private: // Returns a thread local default constructed TensorPrinter static SmartTensorPrinter& DefaultTensorPrinter(); TensorPrinter tensorPrinter_; }; }