mirror of
https://github.com/pytorch/pytorch.git
synced 2025-10-20 12:54:11 +08:00
Replace direct include of caffe2.pb.h with an intermediary header caffe2_pb.h (#10946)
Summary: Pull Request resolved: https://github.com/pytorch/pytorch/pull/10946 ``` codemod -d . --extensions cc,cpp,cu,cuh,h caffe2/proto/caffe2.pb.h caffe2/proto/caffe2_pb.h ``` Reviewed By: houseroad Differential Revision: D9539945 fbshipit-source-id: 497d04720e8e7e61c05ffe1b23733d0cb774de7e
This commit is contained in:
committed by
Facebook Github Bot
parent
5ed62ea6fa
commit
91797c0672
@ -16,7 +16,7 @@
|
||||
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/proto/caffe2_legacy.pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
|
@ -16,7 +16,7 @@
|
||||
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
CAFFE2_DEFINE_string(input_db, "", "The input db.");
|
||||
|
@ -30,7 +30,7 @@
|
||||
#include <string>
|
||||
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "leveldb/db.h"
|
||||
#include "leveldb/write_batch.h"
|
||||
|
@ -30,7 +30,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
CAFFE2_DEFINE_string(input_folder, "", "The input folder name.");
|
||||
|
@ -39,7 +39,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
CAFFE2_DEFINE_bool(shuffle, false,
|
||||
|
@ -24,7 +24,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
CAFFE2_DEFINE_string(image_file, "", "The input image file name.");
|
||||
|
@ -20,7 +20,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/context_gpu.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#define PRINT_SIZE(cls) \
|
||||
std::cout << "Size of " #cls ": " << sizeof(cls) << " bytes." \
|
||||
|
@ -16,7 +16,7 @@
|
||||
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
|
@ -18,7 +18,7 @@
|
||||
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
|
@ -22,7 +22,7 @@
|
||||
#ifdef CAFFE2_OPTIMIZER
|
||||
#include "caffe2/opt/optimizer.h"
|
||||
#endif
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/string_utils.h"
|
||||
|
||||
|
@ -19,7 +19,7 @@
|
||||
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
|
||||
CAFFE2_DEFINE_string(input_db, "", "The input db.");
|
||||
|
@ -21,7 +21,7 @@
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
CAFFE2_DEFINE_string(f_in, "", "The input data file name.");
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/onnx/onnx_exporter.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "onnx/onnx_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -12,7 +12,7 @@
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/typeid.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/blob.h"
|
||||
#include "caffe2/core/common_gpu.h"
|
||||
#include "caffe2/core/context_gpu.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -5,7 +5,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -15,7 +15,7 @@
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
CAFFE2_DEFINE_int64(caffe2_test_big_tensor_size, 100000000, "");
|
||||
|
@ -10,7 +10,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#ifndef CAFFE2_USE_CUDNN
|
||||
#error("This Caffe2 install is not built with cudnn, so you should not include this file.");
|
||||
|
@ -11,7 +11,7 @@
|
||||
#include "caffe2/core/event.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/typeid.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#include "ATen/core/ATenCoreTest.h"
|
||||
#include "ATen/core/ArrayRef.h"
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/event.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/typeid.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -12,7 +12,7 @@
|
||||
#include "caffe2/core/numa.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
// Since we are using the macro CAFFE2_USE_CUDNN, we will need to include this
|
||||
// file after common.h is included.
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -5,7 +5,7 @@
|
||||
|
||||
#include "caffe2/core/blob_serialization.h"
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
namespace db {
|
||||
|
@ -1,7 +1,7 @@
|
||||
#pragma once
|
||||
|
||||
#include "caffe2/core/dispatch/DispatchKey.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/Array.h"
|
||||
#include "caffe2/utils/Metaprogramming.h"
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -1,7 +1,7 @@
|
||||
#pragma once
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/string_utils.h"
|
||||
|
||||
|
@ -23,7 +23,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#define MIOPEN_VERSION 1399
|
||||
|
||||
|
@ -12,7 +12,7 @@
|
||||
#include "caffe2/core/numa.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -24,7 +24,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
#include "caffe2/core/hip/context_hip.h"
|
||||
|
@ -9,7 +9,7 @@
|
||||
|
||||
#include <ATen/core/Error.h>
|
||||
#include "caffe2/core/flags.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
// CAFFE2_LOG_THRESHOLD is a compile time flag that would allow us to turn off
|
||||
// logging at compile time so no logging message below that level is produced
|
||||
|
@ -5,7 +5,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
namespace memonger {
|
||||
|
@ -8,7 +8,7 @@
|
||||
#include "caffe2/core/init.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/string_utils.h"
|
||||
|
||||
|
@ -17,7 +17,7 @@
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/simple_queue.h"
|
||||
#include "caffe2/utils/thread_pool.h"
|
||||
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/stats.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/thread_pool.h"
|
||||
|
||||
|
@ -8,7 +8,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
#include "caffe2/core/context_gpu.h"
|
||||
|
@ -4,7 +4,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/net_dag.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/thread_name.h"
|
||||
|
||||
|
@ -21,7 +21,7 @@
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/simple_queue.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -8,7 +8,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -19,7 +19,7 @@
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/simple_queue.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
CAFFE2_DEFINE_bool(
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -10,7 +10,7 @@
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
#include "caffe2/utils/string_utils.h"
|
||||
|
||||
|
@ -19,7 +19,7 @@
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/operator_schema.h"
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -12,7 +12,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/filler.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -9,7 +9,7 @@
|
||||
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
CAFFE2_DEFINE_bool(
|
||||
caffe2_handle_executor_threads_exceptions,
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
namespace int8 {
|
||||
|
@ -4,7 +4,7 @@
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/graph.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -8,7 +8,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/typeid.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/plan_executor.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
CAFFE2_DEFINE_bool(
|
||||
caffe2_print_blob_sizes_at_exit,
|
||||
|
@ -14,7 +14,7 @@
|
||||
#include "caffe2/core/blob.h"
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/signal_handler.h"
|
||||
#include "caffe2/utils/threadpool/ThreadPool.h"
|
||||
|
||||
|
@ -6,7 +6,7 @@
|
||||
#include "caffe2/core/blob_serialization.h"
|
||||
#include "caffe2/core/db.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -4,7 +4,7 @@
|
||||
#include <caffe2/core/context.h>
|
||||
#include <caffe2/core/operator.h>
|
||||
#include <caffe2/ideep/ideep_utils.h>
|
||||
#include <caffe2/proto/caffe2.pb.h>
|
||||
#include <caffe2/proto/caffe2_pb.h>
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include <ideep.hpp>
|
||||
#include <caffe2/core/operator.h>
|
||||
#include <caffe2/proto/caffe2.pb.h>
|
||||
#include <caffe2/proto/caffe2_pb.h>
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -1,6 +1,6 @@
|
||||
#include <caffe2/core/event_cpu.h>
|
||||
#include <caffe2/core/operator.h>
|
||||
#include <caffe2/proto/caffe2.pb.h>
|
||||
#include <caffe2/proto/caffe2_pb.h>
|
||||
#include <ideep_pin_singletons.hpp>
|
||||
#include "ideep_context.h"
|
||||
|
||||
|
@ -1,5 +1,5 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
CAFFE2_DEFINE_bool(
|
||||
caffe2_mkl_memonger_in_use,
|
||||
|
@ -2,7 +2,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/mkl/mkl_utils.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/math.h"
|
||||
|
||||
#include <gtest/gtest.h>
|
||||
|
@ -5,7 +5,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/mkl/mkl_utils.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#ifdef CAFFE2_HAS_MKL_DNN
|
||||
namespace caffe2 {
|
||||
|
@ -4,7 +4,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/mkl/utils/mkl_dnn_cppwrapper.h"
|
||||
#include "caffe2/mkl/utils/mkl_memory.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
CAFFE2_DECLARE_bool(caffe2_mkl_memonger_in_use);
|
||||
|
||||
|
@ -10,7 +10,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/static_tracepoint.h"
|
||||
#include "caffe2/core/timer.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/registry.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -1,5 +1,5 @@
|
||||
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
void testOpenGL();
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/onnx/backend_rep.h"
|
||||
#include "caffe2/onnx/device.h"
|
||||
#include "caffe2/onnx/helper.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "onnx/onnx_pb.h"
|
||||
|
||||
#include <functional>
|
||||
|
@ -1,7 +1,7 @@
|
||||
#pragma once
|
||||
|
||||
#include "caffe2/predictor/predictor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#include <memory>
|
||||
#include <string>
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/onnx/helper.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "onnx/onnx_pb.h"
|
||||
|
||||
#include <string>
|
||||
|
@ -9,7 +9,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
CAFFE2_DECLARE_bool(caffe2_workspace_stack_debug);
|
||||
|
||||
|
@ -10,7 +10,7 @@
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/operators/create_scope_op.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -1,6 +1,6 @@
|
||||
#include "caffe2/operators/flexible_top_k.h"
|
||||
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -5,7 +5,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/context_gpu.h"
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
#include "caffe2/core/blob.h"
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/math.h"
|
||||
#include <gtest/gtest.h>
|
||||
|
||||
|
@ -6,7 +6,7 @@
|
||||
#include <utility>
|
||||
#include <vector>
|
||||
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/math.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#include <functional>
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "nomnigraph/Graph/Graph.h"
|
||||
#include "nomnigraph/Representations/ControlFlow.h"
|
||||
#include "nomnigraph/Representations/NeuralNet.h"
|
||||
|
@ -11,7 +11,7 @@
|
||||
#include "caffe2/core/operator.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/onnx/onnxifi_init.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
namespace onnx {
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "nomnigraph/Representations/NeuralNet.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -2,7 +2,7 @@
|
||||
#define CAFFE2_OPT_OPTIMIZER_H
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
#include "nomnigraph/Representations/NeuralNet.h"
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
#define CAFFE2_OPT_SINK_H_
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "nomnigraph/Representations/NeuralNet.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/blob.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/proto/predictor_consts.pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
|
2
caffe2/proto/caffe2_pb.h
Normal file
2
caffe2/proto/caffe2_pb.h
Normal file
@ -0,0 +1,2 @@
|
||||
#pragma once
|
||||
#include <caffe2/proto/caffe2.pb.h>
|
@ -12,7 +12,7 @@
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/core/workspace.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/python/pybind_state_dlpack.h"
|
||||
|
||||
#include <pybind11/pybind11.h>
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/core/types.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/python/dlpack.h"
|
||||
|
||||
#include <pybind11/pybind11.h>
|
||||
|
@ -2,7 +2,7 @@
|
||||
#include <stdint.h>
|
||||
#include <zstd.h>
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/transform.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -1,7 +1,7 @@
|
||||
#pragma once
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/transforms/single_op_transform.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/transform.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -3,7 +3,7 @@
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/core/net.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
@ -2,7 +2,7 @@
|
||||
|
||||
#include "caffe2/core/common.h"
|
||||
#include "caffe2/core/transform.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/proto_utils.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
@ -4,7 +4,7 @@
|
||||
#include "caffe2/core/hip/context_hip.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/operators/utility_ops.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/conversions.h"
|
||||
#include "caffe2/utils/math.h"
|
||||
|
||||
|
@ -7,7 +7,7 @@
|
||||
#include "caffe2/core/blob.h"
|
||||
#include "caffe2/core/context.h"
|
||||
#include "caffe2/core/tensor.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
#include "caffe2/utils/conversions.h"
|
||||
#include "caffe2/utils/math.h"
|
||||
|
||||
|
@ -9,7 +9,7 @@
|
||||
|
||||
#include "caffe2/core/logging.h"
|
||||
#include "caffe2/utils/proto_wrap.h"
|
||||
#include "caffe2/proto/caffe2.pb.h"
|
||||
#include "caffe2/proto/caffe2_pb.h"
|
||||
|
||||
namespace caffe2 {
|
||||
|
||||
|
Reference in New Issue
Block a user