Skip to content
This repository has been archived by the owner on Nov 17, 2023. It is now read-only.

Commit

Permalink
fix lint
Browse files Browse the repository at this point in the history
  • Loading branch information
ElaineBao committed Aug 21, 2019
1 parent 201d869 commit f0a5a7f
Showing 1 changed file with 1 addition and 1 deletion.
2 changes: 1 addition & 1 deletion src/operator/nn/mkldnn/mkldnn_batch_norm-inl.h
Original file line number Diff line number Diff line change
Expand Up @@ -214,7 +214,7 @@ void MKLDNNBatchNormForward(const OpContext &ctx, const BatchNormParam &param,
const std::vector<NDArray> &out_data,
const std::vector<NDArray> &aux_states) {
TmpMemMgr::Get()->Init(ctx.requested[batchnorm::kTempSpace]);
unsigned flags = _GetFlags(in_data, aux_states, param, ctx.is_train && !param.use_global_stats);
unsigned flags = _GetFlags(in_data, aux_states, param, ctx.is_train && !param.use_global_stats);
const NDArray &data = in_data[batchnorm::kData];

auto &fwd = GetBNForward<DType>(param, ctx, data, flags);
Expand Down

0 comments on commit f0a5a7f

Please sign in to comment.