Method Record

oneflow/core/framework/tensor_name_scope.cpp:46–55

}

void TensorNameScope::Record(const Tensor* tensor, const std::string& name) {
  uint64_t key = reinterpret_cast<uint64_t>(tensor);
  auto* tensor_names = [&]() {
    if (tensor->is_lazy()) { return &lazy_tensor_names_; }
    return &eager_tensor_names_;
  }();
  std::lock_guard<std::mutex> lock(mutex_);
  // We assume that the name of the tensor will be update more than once.
  (*tensor_names)[key] = name;
}

void TensorNameScope::Record(const std::shared_ptr<Tensor>& tensor, const std::string& name) {
  Record(tensor.get(), name);