reid from https://github.com/michuanhaohao/reid-strong-baseline
zhangmeng
2020-01-17 f7c4a3cfd07adede3308f8d9d3d7315427d90a7c
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