SetPlacement
Description
Sets the placement of a tensor.
Prototype
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
Restrictions
None
Example
Tensor 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: Tensor