SetPlacement

Function Usage

Sets the position for storing tensor data.

Prototype

1

void SetPlacement(Placement placement)

Parameters

1
2
3
4

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

Returns

None

Exception Handling

None

Constraints

None