1 #include "caffe2/operators/expand_op.h" 3 #include "caffe2/core/context_gpu.h" 7 REGISTER_CUDA_OPERATOR(
10 TensorTypes<std::int32_t, std::int64_t, float, double>,
12 REGISTER_CUDA_OPERATOR(
15 TensorTypes<std::int32_t, std::int64_t, float, double>,
A global dictionary that holds information about what Caffe2 modules have been loaded in the current ...