OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-06 12:20:52 +01:00
Code Issues Actions 37 Packages Projects Releases Wiki Activity

Export box_cox operator in caffe2

Browse Source 
Summary: Export box_cox operator in caffe2

Test Plan: Pass all unit tests

Reviewed By: mingzhe09088

Differential Revision: D21515797

fbshipit-source-id: 777ee5e273caeab671ee2c22d133d3f628fb4a6e
This commit is contained in:
Zhuobo Feng 2020-06-17 19:27:31 -07:00 committed by Facebook GitHub Bot
parent 1800032712
commit fb02007e9f
2 changed files with 8 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

5
caffe2/operators/batch_box_cox_op.cc
View File

@ -340,3 +340,8 @@ the transform of each column `x` of the input `data`:
GRADIENT_NOT_IMPLEMENTED_YET(BatchBoxCox);
} // namespace
} // namespace caffe2
C10_EXPORT_CAFFE2_OP_TO_C10_CPU(
BatchBoxCox,
"_caffe2::BatchBoxCox(Tensor data, Tensor lambda1, Tensor lambda2, int min_block_size = 256) -> Tensor results",
caffe2::BatchBoxCoxOp<caffe2::CPUContext>);

3
caffe2/operators/batch_box_cox_op.h
View File

@ -2,10 +2,13 @@
#define CAFFE_OPERATORS_BATCH_BOX_COX_OPS_H_
#include "caffe2/core/context.h"
#include "caffe2/core/export_caffe2_op_to_c10.h"
#include "caffe2/core/logging.h"
#include "caffe2/core/operator.h"
#include "caffe2/utils/math.h"
C10_DECLARE_EXPORT_CAFFE2_OP_TO_C10(BatchBoxCox);
namespace caffe2 {
template <class Context>