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
13
13
cuda7.5_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda75cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz >`_
14
14
cuda8.0_cudnn5_avx_mkl `fluid.tgz <https://guest:@paddleci.ngrok.io/repository/download/Manylinux1_Cuda80cudnn5cp27cp27mu/.lastSuccessful/fluid.tgz >`_
15
15
cuda8.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 >`_
16
17
====================== ========================================
17
18
18
19
从源码编译
Original file line number Diff line number Diff line change @@ -21,8 +21,6 @@ namespace operators {
21
21
22
22
using batch_norm_bwd = mkldnn::batch_normalization_backward;
23
23
using batch_norm_fwd = mkldnn::batch_normalization_forward;
24
- using framework::DataLayout;
25
- using framework::Tensor;
26
24
using mkldnn::memory;
27
25
using mkldnn::primitive;
28
26
using mkldnn::reorder;
@@ -31,18 +29,6 @@ using paddle::platform::MKLDNNDeviceContext;
31
29
using paddle::platform::MKLDNNMemDesc;
32
30
using platform::to_void_cast;
33
31
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
-
46
32
namespace {
47
33
template <typename T>
48
34
struct bn_type_traits {
Original file line number Diff line number Diff line change @@ -22,22 +22,6 @@ limitations under the License. */
22
22
namespace paddle {
23
23
namespace operators {
24
24
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
-
41
25
class BatchNormOp : public framework ::OperatorWithKernel {
42
26
public:
43
27
using framework::OperatorWithKernel::OperatorWithKernel;
Original file line number Diff line number Diff line change @@ -19,6 +19,22 @@ limitations under the License. */
19
19
namespace paddle {
20
20
namespace operators {
21
21
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
+
22
38
template <typename DeviceContext, typename T>
23
39
class BatchNormKernel : public framework ::OpKernel<T> {
24
40
public:
You can’t perform that action at this time.
0 commit comments