SetPlacement
Function Usage
Sets the placement of a tensor.
Prototype
1 | void SetPlacement(const TensorPlacement placement) |
Parameters
Parameter |
Input/Output |
Description |
|---|---|---|
placement |
Input |
Tensor placement to be set. For details about the definition of the TensorPlacement type, see TensorPlacement. |
Returns
None
Constraints
None
Examples
1 2 3 4 5 6 7 | TensorV2 tensor{{{8, 3, 224, 224}, {16, 3, 224, 224}}, // shape {ge::FORMAT_ND, ge::FORMAT_FRACTAL_NZ, {}}, // format kFollowing, // placement ge::DT_FLOAT16, //dt nullptr}; tensor.SetPlacement(TensorPlacement::kOnHost); auto placement = tensor.GetPlacement(); |
Parent topic: TensorV2