TensorProto tensor_proto;
    tensor.AsProtoTensorContent(&tensor_proto);
    return tensor_proto;
  }

  void ConvertOutputTensorToProto(
      std::vector<std::pair<string, Tensor>>& inputs,
      std::vector<std::pair<string, TensorProto>>& protos) {
    for (const auto& input : inputs) {
      TensorProto tensor_proto;
      input.second.AsProtoTensorContent(&tensor_proto);

  tensor.matrix<tsl::qint8>()(0, 1) = tsl::qint8(2);
  tensor.matrix<tsl::qint8>()(1, 0) = tsl::qint8(3);
  tensor.matrix<tsl::qint8>()(1, 1) = tsl::qint8(4);

  ::tensorflow::TensorProto tensor_proto;
  tensor.AsProtoTensorContent(&tensor_proto);
  return ::tensorflow::mangling_util::MangleTensor(tensor_proto);
}

std::string GetQint16Tensor() {
  ::tensorflow::Tensor tensor(::tensorflow::DT_QINT16, {2, 2});

                                   expected.data(), 42);

  // Serialize it to a Tensorproto
  TensorProto proto;
  if (tensorproto_use_tensor_content) {
    expected.AsProtoTensorContent(&proto);
  } else {
    expected.AsProtoField(&proto);
  }

  // Revival should succeed w/o errors
  std::unique_ptr<Constant> revived;

  }

  void Compute(OpKernelContext* ctx) override {
    if (!enabled_) return;

    const Tensor& tensor_data = ctx->input(0);

    TensorProto tensor_proto;
    tensor_data.AsProtoTensorContent(&tensor_proto);

    OP_REQUIRES_OK(ctx, tensor_data_writer_->WriteRecord(
                            tensor_proto.SerializeAsString()));
  }

 private:
  bool enabled_;