message: Fix a memory leak in fused_batch_norm_op.
diff: @@ -947,11 +947,10 @@ class FusedBatchNormGradOpBase : public OpKernel {
  nullptr;
  #if CUDNN_VERSION >= 7402
  + functor::CudnnBatchNormAllocatorInTemp<uint8> workspace_allocator(context);
  if (use_reserved_space) {
  const Tensor& reserve_space = context->input(5);
  reserve_space_data = &reserve_space;
  - functor::CudnnBatchNormAllocatorInTemp<uint8> workspace_allocator(
  - context);
  workspace_allocator_ptr = &workspace_allocator;
  }
  #endif // CUDNN_VERSION >= 7402
  