return err
	}
	return nil
}

func (r *record) content() []byte {
	return r.buf[:r.h.ContentLength]
}

// writeRecord writes and sends a single record.
func (c *conn) writeRecord(recType recType, reqId uint16, b []byte) error {
	c.mutex.Lock()
	defer c.mutex.Unlock()
	c.buf.Reset()
	c.h.init(recType, reqId, len(b))

    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_;
  std::unique_ptr<tsl::WritableFile> tensor_data_file_;