Searched refs:ExampleQuerySpec (Results 1 – 8 of 8) sorted by relevance
51 using ::google::internal::federated::plan::ExampleQuerySpec;70 std::tuple<std::string, ExampleQuerySpec::OutputVectorSpec>>71 GetOutputVectorSpecs(const ExampleQuerySpec::ExampleQuery& example_query) { in GetOutputVectorSpecs()73 std::string, std::tuple<std::string, ExampleQuerySpec::OutputVectorSpec>> in GetOutputVectorSpecs()84 const ExampleQuerySpec::OutputVectorSpec& output_vector_spec, in CheckOutputVectorDataType()85 const ExampleQuerySpec::OutputVectorSpec::DataType& expected_data_type) { in CheckOutputVectorDataType()98 const ExampleQuerySpec& example_query_spec) { in WriteCheckpoint()104 const ExampleQuerySpec::ExampleQuery& example_query = in WriteCheckpoint()109 ExampleQuerySpec::OutputVectorSpec output_vector_spec = in WriteCheckpoint()120 output_vector_spec, ExampleQuerySpec::OutputVectorSpec::INT32)); in WriteCheckpoint()[all …]
57 using ::google::internal::federated::plan::ExampleQuerySpec;182 ExampleQuerySpec::OutputVectorSpec string_vector_spec; in Initialize()185 ExampleQuerySpec::OutputVectorSpec::STRING); in Initialize()186 ExampleQuerySpec::OutputVectorSpec int_vector_spec; in Initialize()188 int_vector_spec.set_data_type(ExampleQuerySpec::OutputVectorSpec::INT64); in Initialize()190 ExampleQuerySpec::ExampleQuery example_query; in Initialize()294 ExampleQuerySpec::OutputVectorSpec float_vector_spec; in TEST_F()296 float_vector_spec.set_data_type(ExampleQuerySpec::OutputVectorSpec::FLOAT); in TEST_F()297 ExampleQuerySpec::OutputVectorSpec string_vector_spec; in TEST_F()302 string_vector_spec.set_data_type(ExampleQuerySpec::OutputVectorSpec::STRING); in TEST_F()[all …]
40 const google::internal::federated::plan::ExampleQuerySpec&
23 // iterators invoked using `ExampleQuerySpec` are expected to return a single
75 // ExampleQuerySpec-based plans, this refers to the total number of row counts
19 google.internal.federated.plan.ExampleQuerySpec.OutputVectorSpec.DataType
389 ExampleQuerySpec example_query_spec = 9 [lazy = true];505 // ExampleQuerySpec message describes client execution that issues example512 message ExampleQuerySpec { message627 // Keys must match the keys in ExampleQuerySpec.output_vector_specs.645 // ExampleQuerySpec) should be placed in an output TF v1 checkpoint.647 // Currently only ExampleQuerySpec output vectors are supported by this663 // Currently only ExampleQuerySpec output vectors are supported by this
1417 example_query_spec: plan_pb2.ExampleQuerySpec, argument1479 example_query_spec: Optional[plan_pb2.ExampleQuerySpec] = None, argument