Searched refs:TFSavedModelAPI (Results 1 – 4 of 4) sorted by relevance
176 Status TFSavedModelAPI::GetFunction(const std::string& function_path, in GetFunction()192 Status TFSavedModelAPI::GetFunctions( in GetFunctions()212 Status TFSavedModelAPI::GetSignatureDefFunction( in GetSignatureDefFunction()233 Status TFSavedModelAPI::GetVariable(const std::string& variable_path, in GetVariable()250 SavedModelV2Bundle* TFSavedModelAPI::GetBundle() { return &this->bundle_; } in GetBundle()252 TFSavedModelAPI::TFSavedModelAPI(const std::string& directory, in TFSavedModelAPI() function in tensorflow::TFSavedModelAPI259 Status TFSavedModelAPI::Load( in Load()262 ImmediateExecutionContext* context, std::unique_ptr<TFSavedModelAPI>* out) { in Load()311 out->reset(new TFSavedModelAPI(directory, std::move(bundle), in Load()
56 class TFSavedModelAPI : public SavedModelAPI {72 std::unique_ptr<TFSavedModelAPI>* out);74 ~TFSavedModelAPI() override = default;81 TFSavedModelAPI(const std::string& directory, SavedModelV2Bundle bundle,
48 std::unique_ptr<tensorflow::TFSavedModelAPI> saved_model; in TF_LoadSavedModel()49 status->status = tensorflow::TFSavedModelAPI::Load( in TF_LoadSavedModel()78 std::unique_ptr<tensorflow::TFSavedModelAPI> saved_model; in TF_LoadSavedModelWithTags()79 status->status = tensorflow::TFSavedModelAPI::Load( in TF_LoadSavedModelWithTags()
505 tensorflow::TFSavedModelAPI* model_api = in TEST_P()506 tensorflow::down_cast<tensorflow::TFSavedModelAPI*>( in TEST_P()