experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Merge pull request #24352 from Intel-tensorflow:relu_build_fix

Browse Source 
PiperOrigin-RevId: 225472163
This commit is contained in:
TensorFlower Gardener 2018-12-13 18:44:48 -08:00
commit 3605ae973e
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
tensorflow/core/kernels/mkl_relu_op.cc
View File

@ -1353,7 +1353,7 @@ class MklLeakyReluOp : public MklReluOpBase<Device, T, eltwise_relu> {
AllocateOutputSetMklShape(context, dst_index, &dst_tensor,
src_tensor.shape(), dnn_shape_dst);
T* out_o = dst_tensor->flat<T>().data();
out_o[0] = user_i[0] >= 0 ? user_g[0] : user_g[0] * this->alpha_;
out_o[0] = user_i[0] >= 0 ? user_i[0] : user_i[0] * this->alpha_;
return;
}
};