reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-02-24 d9b98e947f7b5aae886ff02ec27bd92aa34feaaa
1
2
3
4
5
6
7
8
9
10
11
12
13
#pragma once
 
#include <c10/macros/Macros.h>
 
// We are working on removing the BUILD_NAMEDTENSOR flag from the codebase.
//
// PyTorch's codegen also uses a similar flag. You can find it in
// - aten/src/ATen/env.py
#if !defined(CAFFE2_IS_XPLAT_BUILD) && (!defined(C10_MOBILE) || defined(FEATURE_TORCH_MOBILE))
#ifndef BUILD_NAMEDTENSOR
#define BUILD_NAMEDTENSOR
#endif
#endif