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/distributed.h (1126B)

#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 {

/// \brief Convert to an NNModule and apply a mapping of
/// tensor names to DeviceOptions to it.
///
/// This *only* applies the map to Declare/Export
/// nodes, which are representationally equivalent to
/// external_input/external_output in caffe2 NetDefs.
///
/// Throws an exception if the passed in blobMap contains
/// blobs that are not present in the NNModule.
TORCH_API nom::repr::NNModule convertToNNModule(
    caffe2::NetDef&,
    std::map<std::string, caffe2::DeviceOption>);

/// Helpers for the convertToNNModule for use
/// if you already have an NNModule.
/// You probably don't want to use these
/// if you can use convertToNNModule instead.
TORCH_API void addBlobDeviceOptions(
    std::map<std::string, caffe2::DeviceOption> blobMap,
    nom::repr::NNModule* nn);
TORCH_API void injectDataEdgeIndicators(nom::repr::NNModule* nn);
TORCH_API void removeDataEdgeIndicators(nom::repr::NNModule* nn);

} // 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 { /// \brief Convert to an NNModule and apply a mapping of /// tensor names to DeviceOptions to it. /// /// This *only* applies the map to Declare/Export /// nodes, which are representationally equivalent to /// external_input/external_output in caffe2 NetDefs. /// /// Throws an exception if the passed in blobMap contains /// blobs that are not present in the NNModule. TORCH_API nom::repr::NNModule convertToNNModule( caffe2::NetDef&, std::map); /// Helpers for the convertToNNModule for use /// if you already have an NNModule. /// You probably don't want to use these /// if you can use convertToNNModule instead. TORCH_API void addBlobDeviceOptions( std::map blobMap, nom::repr::NNModule* nn); TORCH_API void injectDataEdgeIndicators(nom::repr::NNModule* nn); TORCH_API void removeDataEdgeIndicators(nom::repr::NNModule* nn); } // namespace caffe2