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:
Edward Yang
2018-08-28 11:40:07 -07:00
committed by Facebook Github Bot
parent 5ed62ea6fa
commit 91797c0672
98 changed files with 99 additions and 97 deletions

View File

@ -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"

View File

@ -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.");

View File

@ -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"

View File

@ -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.");

View File

@ -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,

View File

@ -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.");

View File

@ -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." \

View File

@ -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"

View File

@ -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"

View File

@ -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"

View File

@ -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.");

View File

@ -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.");

View File

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

View File

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

View File

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

View File

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

View File

@ -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, "");

View File

@ -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.");

View 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"

View File

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

View File

@ -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.

View File

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

View File

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

View File

@ -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"

View File

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

View File

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

View File

@ -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"

View File

@ -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

View File

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

View File

@ -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"

View File

@ -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

View File

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

View File

@ -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"

View File

@ -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"

View File

@ -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"

View File

@ -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"

View File

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

View File

@ -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"

View File

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

View File

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

View File

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

View File

@ -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(

View File

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

View File

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

View File

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

View File

@ -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"

View File

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

View File

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

View File

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

View File

@ -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,

View File

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

View File

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

View File

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

View File

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

View File

@ -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,

View File

@ -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"

View File

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

View File

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

View File

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

View File

@ -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"

View File

@ -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,

View File

@ -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>

View File

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

View File

@ -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);

View File

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

View File

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

View File

@ -1,5 +1,5 @@
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
namespace caffe2 {
void testOpenGL();

View File

@ -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>

View File

@ -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>

View File

@ -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>

View File

@ -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);

View File

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

View File

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

View File

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

View File

@ -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>

View File

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

View File

@ -2,7 +2,7 @@
#include "caffe2/core/common.h"
#include "caffe2/proto/caffe2.pb.h"
#include "caffe2/proto/caffe2_pb.h"
#include <functional>

View File

@ -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"

View File

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

View File

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

View File

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

View File

@ -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"

View File

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

View File

@ -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
View File

@ -0,0 +1,2 @@
#pragma once
#include <caffe2/proto/caffe2.pb.h>

View File

@ -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>

View File

@ -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>

View File

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

View File

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

View File

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

View File

@ -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"

View File

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

View File

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

View File

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

View File

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

View File

@ -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"

View File

@ -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"

View File

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