SetBf16Data

Function Usage

Initializes the tensor data by using a memory block of the bfloat16 type for the host tensor allocated by AllocHostTensor.

Prototype

void SetBf16Data(const op::bfloat16 *value, uint64_t size, op::DataType dataType)

Parameters

Parameter	Input/Output	Description
value	Input	Data memory pointer to be written to the aclTensor.
size	Input	Number of elements to be written.
dataType	Input	The data type is . The data is converted into a specified data type and then written to aclTensor.

Returns

None

Constraints

The input parameter pointer must not be null.

Examples

// Initialize a bf16 memory block and assign the value to the first 10 elements of the input.
void Func(const aclTensor *input) {
    bfloat16 myArray[10];
    input->SetData(myArray, 10, DT_BF16);
}

Parent topic: common_types