Fallback to CPU concat op to handle TensorCPU inputs (#15263)
authorGu, Jinghui <jinghui.gu@intel.com>
Mon, 7 Jan 2019 19:07:51 +0000 (11:07 -0800)
committerFacebook Github Bot <facebook-github-bot@users.noreply.github.com>
Mon, 7 Jan 2019 19:13:23 +0000 (11:13 -0800)
commit2ebeb33697d7fb9acaac026cfa140af56e617d11
tree9c576c9f489b02b32f5d0b6a844b51ac827ec1d9
parentc68eb5ec442c4e6cd2ad1a443ac8128b3cd99312
Fallback to CPU concat op to handle TensorCPU inputs (#15263)

Summary:
Fallback to CPU concat op to handle TensorCPU inputs
Pull Request resolved: https://github.com/pytorch/pytorch/pull/15263

Differential Revision: D13587030

Pulled By: yinghai

fbshipit-source-id: 010a8579d61c3beb8556eb92493a552b2ab0030c
caffe2/ideep/operators/concat_split_op.cc
caffe2/python/ideep/concat_split_op_test.py