Searched refs:example_query (Results 1 – 5 of 5) sorted by relevance
190 ExampleQuerySpec::ExampleQuery example_query; in Initialize() local191 example_query.mutable_example_selector()->set_collection_uri( in Initialize()193 (*example_query.mutable_output_vector_specs())[kOutputStringTensorName] = in Initialize()195 (*example_query.mutable_output_vector_specs())[kOutputIntTensorName] = in Initialize()200 ->Add(std::move(example_query)); in Initialize()409 ExampleQuerySpec::ExampleQuery example_query = in TEST_F() local411 (*example_query.mutable_output_vector_specs())["new_tensor"] = in TEST_F()419 ->Add(std::move(example_query)); in TEST_F()465 ExampleQuerySpec::ExampleQuery example_query = in TEST_F() local467 (*example_query.mutable_output_vector_specs())["new_tensor"] = in TEST_F()[all …]
71 GetOutputVectorSpecs(const ExampleQuerySpec::ExampleQuery& example_query) { in GetOutputVectorSpecs() argument76 example_query.output_vector_specs()) { in GetOutputVectorSpecs()104 const ExampleQuerySpec::ExampleQuery& example_query = in WriteCheckpoint() local107 GetOutputVectorSpecs(example_query)) { in WriteCheckpoint()196 for (const auto& example_query : example_query_spec.example_queries()) { in RunPlan() local197 ExampleSelector selector = example_query.example_selector(); in RunPlan()
137 for (const auto& example_query : in RunPlanWithExampleQuerySpec() local140 example_query.output_vector_specs()) { in RunPlanWithExampleQuerySpec()
517 for (const auto& example_query : in RunPlanWithExampleQuerySpec() local520 example_query.output_vector_specs()) { in RunPlanWithExampleQuerySpec()
1440 for example_query in example_query_spec.example_queries:1441 vector_names = set(example_query.output_vector_specs.keys())