SetFp16Data

Description

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 data type before it is written to the aclTensor.

For details about ge::DataType, see ge::DataType in Basic Data Structures and APIs.

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->SetFp16Data(myArray, 10, DT_FLOAT16);
}