Class InferenceMemoryNLP

• Defined in File inference_memory_nlp.hpp

Base Type

• public morpheus::InferenceMemory (Class InferenceMemory)

class InferenceMemoryNLP : public morpheus::InferenceMemory

This is a container class for data that needs to be submitted to the inference server for NLP category usecases.

Public Functions

InferenceMemoryNLP(std::size_t count, TensorObject input_ids, TensorObject input_mask, TensorObject seq_ids)

Construct a new Inference Memory NLP object.

Parameters

• count – : Number of messages

• input_ids – : The token-ids for each string padded with 0s to max_length

• input_mask – : The mask for token-ids result where corresponding positions identify valid token-id values

• seq_ids – : Ids used to index from an inference input to a message. Necessary since there can be more inference inputs than messages (i.e., if some messages get broken into multiple inference requests)

const TensorObject &get_input_ids() const

Get the input ids object.

Returns

const TensorObject&

const TensorObject &get_input_mask() const

Get the input mask object.

Returns

const TensorObject&

const TensorObject &get_seq_ids() const

Get the seq ids object.

Returns

const TensorObject&

void set_input_ids(TensorObject input_ids)

Set the input ids object.

Parameters

input_ids –

void set_input_mask(TensorObject input_mask)

Set the input mask object.

Parameters

input_mask –

void set_seq_ids(TensorObject seq_ids)

Set the seq ids object.

Parameters

seq_ids –