Method operator()

oneflow/core/functional/impl/array_functor.cpp:3160–3182

class To2Functor {
 public:
  Maybe<Tensor> operator()(const std::shared_ptr<Tensor>& input,
                           const Optional<Symbol<Device>>& device_,
                           const Optional<Symbol<DType>>& dtype_, bool copy) const {
    if (input->is_global()) {
      if (!device_.has_value()) {
        std::string device_type = JUST(input->parallel_desc())->device_tag();
        return JUST(GlobalTensorTo(input, device_type, dtype_.value_or(input->dtype()), copy));
      } else {
        if (!GlobalMode::is_enabled()) {
          CHECK_OR_RETURN(!device_.has_value())
              << Error::RuntimeError()
              << "Only string device without device id (eg. \"cpu\" or \"cuda\") is expected "
              << "for global tensor, but got " << device_.value_or(Symbol<Device>())->ToRepr();
        }
        std::string device_type = device_.value_or(Symbol<Device>())->type();
        return JUST(GlobalTensorTo(input, device_type, dtype_.value_or(input->dtype()), copy));
      }
    } else {
      auto dtype = dtype_.value_or(input->dtype());
      auto device = device_.value_or(JUST(input->device()));
      return JUST(LocalTensorTo(input, device, dtype, copy));
    }
  }
};

class To3Functor {