TensorPlacementUtils

class TensorPlacementUtils {
 public:
  // 判断Tensor是否位于Device上的内存
  static bool IsOnDevice(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Host上
  static bool IsOnHost(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Host上，且数据紧跟在结构体后面
  static bool IsOnHostFollowing(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Host上，且数据不紧跟在结构体后面
  static bool IsOnHostNotFollowing(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Device上的内存
  static bool IsOnDeviceHbm(TensorPlacement placement) {
    ...
  }
  // 判断Tensor是否位于Device上的P2p内存
  static bool IsOnDeviceP2p(TensorPlacement placement) {
    ...
  }
};

TensorData tensor_data;
tensor_data.SetPlacement(TensorPlacement::kOnHost);
auto on_host = TensorPlacementUtils::IsOnHost(tensor_data.GetPlacement()); // on_host is true