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/backend_transformer_base.h (2773B)

#pragma once

#include "caffe2/core/common.h"
#include "caffe2/core/workspace.h"
#include "caffe2/opt/bound_shape_inferencer.h"
#include "caffe2/proto/caffe2_pb.h"

#include <string>
#include <unordered_map>
#include <vector>

namespace caffe2 {
namespace {
constexpr char kNetPos[] = "net_pos";
constexpr char kModelId[] = "model_id";
} // namespace

struct BackendTransformOptions {
  explicit BackendTransformOptions() : bound_shape_spec(0, 0) {}

// Enable debugging by dumping more intermediate graphs
  bool debug{false};

// Minimum number of ops to create a backend op. If the subgraph is too
  // small, it doesn't make sense to lower it to backend.
  size_t min_ops{1};

// Bound shape spec
  BoundShapeSpec bound_shape_spec;
};

// Wrap TensorShape into TensorProto
TensorProto wrapShapeInfoIntoTensorProto(
    const std::string& name,
    const ShapeInfo& shape_info);

// Wrap Quantized TensorShape into QTensorProto
QTensorProto wrapShapeInfoIntoQTensorProto(
    const std::string& name,
    const ShapeInfo& shape_info);

// This class contains some common functions for backend lowering and graph
// cutting
class BackendTransformerBase {
 public:
  BackendTransformerBase() {}
  virtual ~BackendTransformerBase() {}

const std::unordered_map<std::string, std::string>& input_mapping() const {
    return input_mapping_;
  }

const std::unordered_map<std::string, std::string>& reverse_input_mapping()
      const {
    return reverse_input_mapping_;
  }

virtual void transform(
      Workspace* ws,
      NetDef* pred_net,
      const std::vector<std::string>& weight_names,
      const ShapeInfoMap& shape_hints,
      const std::unordered_set<int>& blocklisted_ops) = 0;

static void annotateOpIndex(NetDef* net);

// Get model ID from the NetDef
  static std::string getModelId(const NetDef& net);

protected:
  // add shape info to the net
  void addShapeToNet(NetDef& shape_net, const ShapeInfoMap& shape_hints) const;

// Dump the net with shape info
  void dumpNet(
      const NetDef& pred_net,
      const ShapeInfoMap& map,
      const std::string& fname) const;

// SSA rewrite the net and return name mapping
  ShapeInfoMap ssaRewriteAndMapNames(
      Workspace* ws,
      NetDef* pred_net,
      const ShapeInfoMap& input_shape_hints);

// Do bound shape inference and collect shape infos
  ShapeInfoMap inferShapes(
      Workspace* ws,
      NetDef* pred_net,
      const ShapeInfoMap& shape_hints_mapped,
      const BoundShapeSpec& spec);

// Input mapping of input name -> original input name
  std::unordered_map<std::string, std::string> input_mapping_;

// Input mapping of original input name -> input name
  std::unordered_map<std::string, std::string> reverse_input_mapping_;
};
} // namespace caffe2

#pragma once #include "caffe2/core/common.h" #include "caffe2/core/workspace.h" #include "caffe2/opt/bound_shape_inferencer.h" #include "caffe2/proto/caffe2_pb.h" #include #include #include namespace caffe2 { namespace { constexpr char kNetPos[] = "net_pos"; constexpr char kModelId[] = "model_id"; } // namespace struct BackendTransformOptions { explicit BackendTransformOptions() : bound_shape_spec(0, 0) {} // Enable debugging by dumping more intermediate graphs bool debug{false}; // Minimum number of ops to create a backend op. If the subgraph is too // small, it doesn't make sense to lower it to backend. size_t min_ops{1}; // Bound shape spec BoundShapeSpec bound_shape_spec; }; // Wrap TensorShape into TensorProto TensorProto wrapShapeInfoIntoTensorProto( const std::string& name, const ShapeInfo& shape_info); // Wrap Quantized TensorShape into QTensorProto QTensorProto wrapShapeInfoIntoQTensorProto( const std::string& name, const ShapeInfo& shape_info); // This class contains some common functions for backend lowering and graph // cutting class BackendTransformerBase { public: BackendTransformerBase() {} virtual ~BackendTransformerBase() {} const std::unordered_map& input_mapping() const { return input_mapping_; } const std::unordered_map& reverse_input_mapping() const { return reverse_input_mapping_; } virtual void transform( Workspace* ws, NetDef* pred_net, const std::vector& weight_names, const ShapeInfoMap& shape_hints, const std::unordered_set& blocklisted_ops) = 0; static void annotateOpIndex(NetDef* net); // Get model ID from the NetDef static std::string getModelId(const NetDef& net); protected: // add shape info to the net void addShapeToNet(NetDef& shape_net, const ShapeInfoMap& shape_hints) const; // Dump the net with shape info void dumpNet( const NetDef& pred_net, const ShapeInfoMap& map, const std::string& fname) const; // SSA rewrite the net and return name mapping ShapeInfoMap ssaRewriteAndMapNames( Workspace* ws, NetDef* pred_net, const ShapeInfoMap& input_shape_hints); // Do bound shape inference and collect shape infos ShapeInfoMap inferShapes( Workspace* ws, NetDef* pred_net, const ShapeInfoMap& shape_hints_mapped, const BoundShapeSpec& spec); // Input mapping of input name -> original input name std::unordered_map input_mapping_; // Input mapping of original input name -> input name std::unordered_map reverse_input_mapping_; }; } // namespace caffe2