| 52 | namespace { |
| 53 | |
| 54 | Status InitializeVariables(Session* session, |
| 55 | const std::vector<string>& init_ops) { |
| 56 | LOG(INFO) << "Initializing graph variables"; |
| 57 | for (const string& init_op : init_ops) { |
| 58 | TF_RETURN_IF_ERROR(session->Run({}, {}, {init_op}, nullptr)); |
| 59 | } |
| 60 | return Status::OK(); |
| 61 | } |
| 62 | |
| 63 | template <class T> |
| 64 | void InitializeTensor(const std::vector<float>& initialization_values, |