Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 6 of 6 for kSavedModelTagServe (0.2 sec)

  1. tensorflow/cc/saved_model/saved_model_bundle_lite_test.cc

          io::JoinPath(testing::TensorFlowSrcRoot(), kTestDataSharded);
      for (int i = 0; i < 100; ++i) {
        TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir,
                                    {kSavedModelTagServe}, &bundle));
        CheckSavedModelBundle(export_dir, bundle);
      }
    }
    
    TEST_F(LoaderTest, ExtendFailsTest) {
      SavedModelBundleLite bundle;
      SessionOptions session_options;
      RunOptions run_options;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 9.7K bytes
    - Viewed (0)
  2. tensorflow/cc/saved_model/saved_model_bundle_test.cc

      TF_ASSERT_OK(LoadSavedModel(session_options, run_options, export_dir,
                                  {kSavedModelTagServe}, &bundle));
      MetaGraphDef actual_metagraph;
      TF_ASSERT_OK(ReadMetaGraphDefFromSavedModel(export_dir, {kSavedModelTagServe},
                                                  &actual_metagraph));
      EXPECT_EQ(actual_metagraph.DebugString(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Apr 14 19:16:58 UTC 2023
    - 15.3K bytes
    - Viewed (0)
  3. tensorflow/cc/saved_model/reader_test.cc

     protected:
      ReaderTest() {}
    
      void CheckMetaGraphDef(const MetaGraphDef& meta_graph_def) {
        const auto& tags = meta_graph_def.meta_info_def().tags();
        EXPECT_TRUE(std::find(tags.begin(), tags.end(), kSavedModelTagServe) !=
                    tags.end());
        EXPECT_NE(meta_graph_def.meta_info_def().tensorflow_version(), "");
        EXPECT_EQ(
            meta_graph_def.signature_def().at("serving_default").method_name(),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed May 17 21:17:08 UTC 2023
    - 5.1K bytes
    - Viewed (0)
  4. tensorflow/cc/saved_model/saved_model_fuzz.cc

      TF_CHECK_OK(tsl::WriteBinaryProto(tensorflow::Env::Default(),
                                        export_dir + kSavedModelFilenamePb, model));
    
      LoadSavedModel(session_options, run_options, export_dir,
                     {kSavedModelTagServe}, &bundle)
          .IgnoreError();
    }
    FUZZ_TEST(SavedModelFuzz, FuzzLoadSavedModel);
    
    }  // namespace
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Feb 09 23:47:46 UTC 2023
    - 1.5K bytes
    - Viewed (0)
  5. tensorflow/cc/saved_model/tag_constants.h

    /// Tag for the `gpu` graph.
    constexpr char kSavedModelTagGpu[] = "gpu";
    
    /// Tag for the `tpu` graph.
    constexpr char kSavedModelTagTpu[] = "tpu";
    
    /// Tag for the `serving` graph.
    constexpr char kSavedModelTagServe[] = "serve";
    
    /// Tag for the `training` graph.
    constexpr char kSavedModelTagTrain[] = "train";
    
    }  // namespace tensorflow
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jan 24 22:31:28 UTC 2018
    - 1.2K bytes
    - Viewed (0)
  6. tensorflow/c/c_api_test.cc

      TF_Buffer* run_options = TF_NewBufferFromString("", 0);
      TF_Buffer* metagraph = TF_NewBuffer();
      TF_Status* s = TF_NewStatus();
      const char* tags[] = {tensorflow::kSavedModelTagServe};
      TF_Graph* graph = TF_NewGraph();
      TF_Session* session = TF_LoadSessionFromSavedModel(
          opt, run_options, saved_model_dir.c_str(), tags, 1, graph, metagraph, s);
      TF_DeleteBuffer(run_options);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
Back to top