Method Set

oneflow/core/framework/variable_tensor_mgr.cpp:27–37

namespace oneflow {

Maybe<void> VariableTensorMgr::Set(const std::string& variable_op_name,
                                   const std::shared_ptr<one::Tensor>& variable_tensor,
                                   const Symbol<DType>& dtype) {
  if (dtype && variable_tensor->dtype() != dtype) {
    LazyMode::Guard guard{false};
    variables_[variable_op_name] = JUST(one::functional::Cast(variable_tensor, dtype, false));
  } else {
    variables_[variable_op_name] = variable_tensor;
  }
  return Maybe<void>::Ok();
}

Maybe<one::Tensor> VariableTensorMgr::Get(const std::string& variable_op_name,
                                          const Symbol<DType>& dtype) {