mirror of
https://github.com/zebrajr/pytorch.git
synced 2025-12-07 12:21:27 +01:00
c44c95fd0b
* operator 'expand' * updated operator with a simple testcase * Revert "updated operator with a simple testcase" This reverts commit 1ce9f8ac567b525677254b0dce5735d7fea133d7. * updated operator with a simple testcase * expand operator with a passed testcase * typo * GPU full support added * GPU support testing... * GPU full supported * formatted * nits repaired * gpu parameters fixed * Expander removed * nits fixed, document added * formatted * new testcases added & nits repaired
18 lines
411 B
C++
18 lines
411 B
C++
#include "caffe2/operators/expand_op.h"
|
|
|
|
#include "caffe2/core/context_gpu.h"
|
|
|
|
namespace caffe2 {
|
|
|
|
REGISTER_CUDA_OPERATOR(
|
|
Expand,
|
|
ExpandOp<
|
|
TensorTypes<std::int32_t, std::int64_t, float, double>,
|
|
CUDAContext>);
|
|
REGISTER_CUDA_OPERATOR(
|
|
ExpandGradient,
|
|
ExpandGradientOp<
|
|
TensorTypes<std::int32_t, std::int64_t, float, double>,
|
|
CUDAContext>);
|
|
} // namespace caffe2
|