The type of tensor z should be bool. And there's no need to call mutable_data because ElementwiseComputeEx will do it.