OSSForks/pytorch
1
0
Fork 0
mirror of https://github.com/zebrajr/pytorch.git synced 2025-12-07 12:21:27 +01:00
Code Issues Actions 38 Packages Projects Releases Wiki Activity

Support fallback for more operators (#16566)

Browse Source 
Summary:
Pull Request resolved: https://github.com/pytorch/pytorch/pull/16566

it's a follow-up to https://github.com/pytorch/pytorch/pull/16456

Reviewed By: yinghai

Differential Revision: D13881462

fbshipit-source-id: eff063580ac8f622477417ed4b25320299451811
This commit is contained in:
Tim Khatkevich 2019-01-30 13:15:59 -08:00 committed by Facebook Github Bot
parent 307c83b5eb
commit 2ed5569bd6
1 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

4
caffe2/ideep/operators/operator_fallback_ideep.cc
View File

@ -19,8 +19,10 @@
#include <caffe2/operators/elementwise_mul_op.h>
#include <caffe2/operators/elementwise_ops.h>
#include <caffe2/operators/expand_op.h>
#include <caffe2/operators/expand_squeeze_dims_op.h>
#include <caffe2/operators/filler_op.h>
#include <caffe2/operators/flatten_op.h>
#include <caffe2/operators/gather_op.h>
#include <caffe2/operators/generate_proposals_op.h>
#include <caffe2/operators/given_tensor_fill_op.h>
#include <caffe2/operators/load_save_op.h>
@ -204,6 +206,8 @@ REGISTER_IDEEP_OPERATOR(
IDEEPFallbackOp<ExpandOp<
TensorTypes<std::int32_t, std::int64_t, float, double>,
CPUContext>>);
REGISTER_IDEEP_OPERATOR(Gather, IDEEPFallbackOp<GatherOp<CPUContext>>);
REGISTER_IDEEP_OPERATOR(ExpandDims, IDEEPFallbackOp<ExpandDimsOp<CPUContext>>);
REGISTER_IDEEP_OPERATOR(
ReduceL2,