File tree Expand file tree Collapse file tree 4 files changed +17
-30
lines changed
doc/fluid/howto/inference Expand file tree Collapse file tree 4 files changed +17
-30
lines changed Original file line number Diff line number Diff line change @@ -13,6 +13,7 @@ cpu_noavx_openblas `fluid.tgz <https://guest:@paddleci.ngrok.io/repository
1313cuda7.5_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda75cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz >`_
1414cuda8.0_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda80cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz >`_
1515cuda8.0_cudnn7_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda8cudnn7cp27cp27mu/.lastSuccessful/fluid.tgz >`_
16+ cuda9.0_cudnn7_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda90cudnn7avxMkl/.lastSuccessful/fluid.tgz >`_
1617====================== ========================================
1718
1819从源码编译
Original file line number Diff line number Diff line change @@ -21,8 +21,6 @@ namespace operators {
2121
2222using batch_norm_bwd = mkldnn::batch_normalization_backward;
2323using batch_norm_fwd = mkldnn::batch_normalization_forward;
24- using framework::DataLayout;
25- using framework::Tensor;
2624using mkldnn::memory;
2725using mkldnn::primitive;
2826using mkldnn::reorder;
@@ -31,18 +29,6 @@ using paddle::platform::MKLDNNDeviceContext;
3129using paddle::platform::MKLDNNMemDesc;
3230using platform::to_void_cast;
3331
34- template <typename T>
35- using EigenArrayMap =
36- Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
37- template <typename T>
38- using ConstEigenArrayMap =
39- Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
40- template <typename T>
41- using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1 >>;
42- template <typename T>
43- using ConstEigenVectorArrayMap =
44- Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1 >>;
45-
4632namespace {
4733template <typename T>
4834struct bn_type_traits {
Original file line number Diff line number Diff line change @@ -22,22 +22,6 @@ limitations under the License. */
2222namespace paddle {
2323namespace operators {
2424
25- using Tensor = framework::Tensor;
26- using LoDTensor = framework::LoDTensor;
27- using DataLayout = framework::DataLayout;
28-
29- template <typename T>
30- using EigenArrayMap =
31- Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
32- template <typename T>
33- using ConstEigenArrayMap =
34- Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
35- template <typename T>
36- using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1 >>;
37- template <typename T>
38- using ConstEigenVectorArrayMap =
39- Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1 >>;
40-
4125class BatchNormOp : public framework ::OperatorWithKernel {
4226 public:
4327 using framework::OperatorWithKernel::OperatorWithKernel;
Original file line number Diff line number Diff line change @@ -19,6 +19,22 @@ limitations under the License. */
1919namespace paddle {
2020namespace operators {
2121
22+ using Tensor = framework::Tensor;
23+ using LoDTensor = framework::LoDTensor;
24+ using DataLayout = framework::DataLayout;
25+
26+ template <typename T>
27+ using EigenArrayMap =
28+ Eigen::Map<Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
29+ template <typename T>
30+ using ConstEigenArrayMap =
31+ Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, Eigen::Dynamic>>;
32+ template <typename T>
33+ using EigenVectorArrayMap = Eigen::Map<Eigen::Array<T, Eigen::Dynamic, 1 >>;
34+ template <typename T>
35+ using ConstEigenVectorArrayMap =
36+ Eigen::Map<const Eigen::Array<T, Eigen::Dynamic, 1 >>;
37+
2238template <typename DeviceContext, typename T>
2339class BatchNormKernel : public framework ::OpKernel<T> {
2440 public:
You can’t perform that action at this time.
0 commit comments