Method AsLocalTensor

oneflow/core/framework/tensor.cpp:59–63

}

Maybe<LocalTensor> StaticZerosTensor::AsLocalTensor() {
  CHECK_OR_RETURN(is_local());  // NOLINT
  return std::dynamic_pointer_cast<LocalTensor>(
      JUST(functional::Constant(*shape_, Scalar(0), CHECK_JUST(DType::Get(dtype_)), device_)));
}

Parameter::Parameter(const std::shared_ptr<Tensor>& tensor, bool requires_grad)
    : ProxyTensor<Parameter>(tensor) {