FM / opt

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

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

Name	Size	Mode	Actions
annotations.h	2141	0644	edit dl rm
backend_cutting.h	586	0644	edit dl rm
backend_transformer_base.h	2773	0644	edit dl rm
bound_shape_inferencer.h	5884	0644	edit dl rm
converter.h	3013	0644	edit dl rm
device.h	442	0644	edit dl rm
distributed.h	1126	0644	edit dl rm
fakefp16_transform.h	584	0644	edit dl rm
fusion.h	4141	0644	edit dl rm
glow_net_transform.h	1625	0644	edit dl rm
mobile.h	388	0644	edit dl rm
onnxifi_op.h	19497	0644	edit dl rm
onnxifi_transformer.h	6859	0644	edit dl rm
onnx_convert.h	1279	0644	edit dl rm
optimizer.h	402	0644	edit dl rm
optimize_ideep.h	351	0644	edit dl rm
passes.h	2642	0644	edit dl rm
shape_info.h	4784	0644	edit dl rm
tvm_transformer.h	2992	0644	edit dl rm

Edit: /usr/local/lib64/python3.6/site-packages/torch/include/caffe2/opt/annotations.h (2141B)

#pragma once

#include "caffe2/core/common.h"
#include "caffe2/core/logging.h"
#include "caffe2/proto/caffe2_pb.h"
#include "nomnigraph/Representations/NeuralNet.h"

namespace caffe2 {

class TORCH_API Caffe2Annotation : public nom::repr::Annotation {
 public:
  Caffe2Annotation() : Annotation(AnnotationKind::Caffe2) {}
  Caffe2Annotation(std::string device)
      : Annotation(AnnotationKind::Caffe2), Device(device) {}
  virtual ~Caffe2Annotation() {}

void setOperatorDef(const caffe2::OperatorDef& opDef);
  bool hasOperatorDef() const;
  const caffe2::OperatorDef& getOperatorDef() const;
  caffe2::OperatorDef* getMutableOperatorDef();

void setDeviceOption(const caffe2::DeviceOption& opDef);
  bool hasDeviceOption() const;
  const caffe2::DeviceOption& getDeviceOption() const;
  caffe2::DeviceOption* getMutableDeviceOption();

// Distributed annotations
  void setDevice(std::string device);
  const std::string getDevice() const;
  void setDeviceType(int device);
  int getDeviceType() const;

enum class ParallelizationScheme {
    none,
    split_by_batch,
    split_by_length,
    shard,
    shard_by_number
  };
  void setParallelization(ParallelizationScheme, int num = -1);
  ParallelizationScheme getParallelizationScheme() const;
  int getParallelization() const;

void setKeyNode(nom::repr::NNGraph::NodeRef);
  const nom::repr::NNGraph::NodeRef& getKeyNode() const;
  void setLengthNode(nom::repr::NNGraph::NodeRef);
  const nom::repr::NNGraph::NodeRef& getLengthNode() const;

void setComponentLevels(std::vector<std::string> components);
  std::vector<std::string> getComponentLevels() const;

static bool classof(const Annotation* A);

private:
  std::string Device = "";
  caffe2::OperatorDef OpDef;
  bool OpDefExists = false;

// Distributed annotations
  int DeviceType = caffe2::DeviceTypeProto::PROTO_CPU;
  ParallelizationScheme parallelization_scheme_ = ParallelizationScheme::none;
  int parallelization_ = -1;
  nom::repr::NNGraph::NodeRef key_node_ = nullptr;
  nom::repr::NNGraph::NodeRef length_node_ = nullptr;
  std::vector<std::string> component_levels_;
};

} // namespace caffe2

#pragma once #include "caffe2/core/common.h" #include "caffe2/core/logging.h" #include "caffe2/proto/caffe2_pb.h" #include "nomnigraph/Representations/NeuralNet.h" namespace caffe2 { class TORCH_API Caffe2Annotation : public nom::repr::Annotation { public: Caffe2Annotation() : Annotation(AnnotationKind::Caffe2) {} Caffe2Annotation(std::string device) : Annotation(AnnotationKind::Caffe2), Device(device) {} virtual ~Caffe2Annotation() {} void setOperatorDef(const caffe2::OperatorDef& opDef); bool hasOperatorDef() const; const caffe2::OperatorDef& getOperatorDef() const; caffe2::OperatorDef* getMutableOperatorDef(); void setDeviceOption(const caffe2::DeviceOption& opDef); bool hasDeviceOption() const; const caffe2::DeviceOption& getDeviceOption() const; caffe2::DeviceOption* getMutableDeviceOption(); // Distributed annotations void setDevice(std::string device); const std::string getDevice() const; void setDeviceType(int device); int getDeviceType() const; enum class ParallelizationScheme { none, split_by_batch, split_by_length, shard, shard_by_number }; void setParallelization(ParallelizationScheme, int num = -1); ParallelizationScheme getParallelizationScheme() const; int getParallelization() const; void setKeyNode(nom::repr::NNGraph::NodeRef); const nom::repr::NNGraph::NodeRef& getKeyNode() const; void setLengthNode(nom::repr::NNGraph::NodeRef); const nom::repr::NNGraph::NodeRef& getLengthNode() const; void setComponentLevels(std::vector components); std::vector getComponentLevels() const; static bool classof(const Annotation* A); private: std::string Device = ""; caffe2::OperatorDef OpDef; bool OpDefExists = false; // Distributed annotations int DeviceType = caffe2::DeviceTypeProto::PROTO_CPU; ParallelizationScheme parallelization_scheme_ = ParallelizationScheme::none; int parallelization_ = -1; nom::repr::NNGraph::NodeRef key_node_ = nullptr; nom::repr::NNGraph::NodeRef length_node_ = nullptr; std::vector component_levels_; }; } // namespace caffe2