SetFp16Data

Function Usage

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

Prototype

void SetFp16Data(const op::fp16_t *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 op::DataType (ge::DataType), which converts data to the specified dataType before it is written to the aclTensor.

Returns

None

Constraints

The input parameter pointer must not be null.

Example

1
2
3
4
5
// Initialize a fp16 memory block and assign the value to the first 10 elements of the input.
void Func(const aclTensor *input) {
    fp16_t myArray[10];
    input->SetData(myArray, 10, DT_FLOAT16);
}