Method DeviceOrHostCPU

tensorflow/core/common_runtime/eager/tensor_handle.cc:286–288

}

Device* TensorHandle::DeviceOrHostCPU(EagerContext* ctx) const {
  return (device_ == nullptr) ? ctx->HostCPU() : device_;
}

Status TensorHandle::Shape(tensorflow::TensorShape* shape) {
  TF_RETURN_IF_ERROR(WaitReady());