MCPcopy Create free account
hub / github.com/Oneflow-Inc/oneflow / FillVariableTensorMgr

Function FillVariableTensorMgr

oneflow/api/common/variable_tensor_mgr.h:25–30  ·  view source on GitHub ↗

Source from the content-addressed store, hash-verified

23namespace oneflow {
24
25inline Maybe<void> FillVariableTensorMgr(
26 const std::vector<std::string>& variable_op_names,
27 const std::vector<std::shared_ptr<one::Tensor>>& variable_tensors) {
28 auto mgr = Singleton<VariableTensorMgr>::Get();
29 return mgr->Fill(variable_op_names, variable_tensors);
30}
31
32inline void ResetVariableTensorMgr() {
33 auto mgr = Singleton<VariableTensorMgr>::Get();

Callers 1

LoadCheckpointMethod · 0.85

Calls 2

GetFunction · 0.85
FillMethod · 0.45

Tested by

no test coverage detected