SetPlacement

Sets the position for storing tensor data.

void SetPlacement(Placement placement)

Parameter

Input/Output

Description

placement

Input

Value of the data address to be set.

The enumerated values are defined as follows:

enum Placement {
  kPlacementHost = 0,   // host data addr
  kPlacementDevice = 1, // device data addr
};

None.

None.

None.

Parent topic: TensorDesc