| /external/federated-compute/fcp/client/engine/ |
| D | common.cc | 37 const TensorflowSpec& tensorflow_spec, in ValidateTensorflowSpec() argument
42 tensorflow_spec.input_tensor_specs_size()) { in ValidateTensorflowSpec()
48 tensorflow_spec.input_tensor_specs()) { in ValidateTensorflowSpec()
58 tensorflow_spec.output_tensor_specs_size()) { in ValidateTensorflowSpec()
62 tensorflow_spec.output_tensor_specs_size())); in ValidateTensorflowSpec()
65 tensorflow_spec.output_tensor_specs()) { in ValidateTensorflowSpec()
|
| D | simple_plan_engine.cc | 56 const TensorflowSpec& tensorflow_spec, const std::string& graph, in RunPlan() argument
67 tensorflow_spec, expected_input_tensor_names_set, output_names); in RunPlan()
87 RunPlanInternal(tf_wrapper.get(), tensorflow_spec, std::move(inputs), in RunPlan()
117 const google::internal::federated::plan::TensorflowSpec& tensorflow_spec, in RunPlanInternal() argument
131 tensorflow_spec.dataset_token_tensor_name(), total_example_count, in RunPlanInternal()
136 tensorflow_spec.target_node_names()) { in RunPlanInternal()
140 !tensorflow_spec.constant_inputs().empty()) { in RunPlanInternal()
143 for (const auto& [name, tensor_proto] : tensorflow_spec.constant_inputs()) { in RunPlanInternal()
|
| D | tflite_plan_engine.cc | 75 const TensorflowSpec& tensorflow_spec, const std::string& model, in RunPlan() argument
86 tensorflow_spec, expected_input_tensor_names_set, output_names); in RunPlan()
97 tensorflow_spec.dataset_token_tensor_name(), &total_example_count, in RunPlan()
101 if (!tensorflow_spec.constant_inputs().empty()) { in RunPlan()
110 tensorflow_spec.constant_inputs()) { in RunPlan()
|
| D | simple_plan_engine.h | 62 const google::internal::federated::plan::TensorflowSpec& tensorflow_spec,
73 const google::internal::federated::plan::TensorflowSpec& tensorflow_spec,
|
| D | common.h | 72 const google::internal::federated::plan::TensorflowSpec& tensorflow_spec,
|
| D | tflite_plan_engine.h | 62 const google::internal::federated::plan::TensorflowSpec& tensorflow_spec,
|
| D | tflite_plan_engine_test.cc | 139 client_only_plan_.phase().tensorflow_spec().output_tensor_specs()) { in InitializeFlTask()
217 client_only_plan_.phase().tensorflow_spec(), in TEST_F()
|
| /external/federated-compute/fcp/artifact_building/ |
| D | plan_utils.py | 125 def convert(graph_def, tensorflow_spec, guarantee_all_funcs_one_use=False): argument
133 graph_def, tensorflow_spec, guarantee_all_funcs_one_use
141 return convert(graph_def, tensorflow_spec, True)
154 client_graph_def, plan.phase[0].client_phase.tensorflow_spec
158 plan.tflite_graph = convert(client_graph_def, plan.phase.tensorflow_spec)
|
| D | plan_utils_test.py | 96 tensorflow_spec = plan_pb2.TensorflowSpec()
97 tensorflow_spec.input_tensor_specs.append(input_tensor_spec)
98 tensorflow_spec.output_tensor_specs.append(output_tensor_spec)
101 graph.as_graph_def(), tensorflow_spec
202 plan_proto.phase[0].client_phase.tensorflow_spec.input_tensor_specs.append(
205 plan_proto.phase[0].client_phase.tensorflow_spec.output_tensor_specs.append(
232 client_only_plan_proto.phase.tensorflow_spec.input_tensor_specs.append(
235 client_only_plan_proto.phase.tensorflow_spec.output_tensor_specs.append(
|
| D | federated_compute_plan_builder.py | 1377 tensorflow_spec = plan_pb2.TensorflowSpec()
1379 tensorflow_spec.dataset_token_tensor_name = token_placeholder.name
1381 tensorflow_spec.input_tensor_specs.extend(
1386 tensorflow_spec.output_tensor_specs.extend(output_tensor_specs)
1388 tensorflow_spec.target_node_names.extend(target_nodes)
1396 tensorflow_spec.constant_inputs[placeholder.name].dtype = 0
1411 tensorflow_spec=tensorflow_spec, federated_compute=io_router
|
| /external/federated-compute/fcp/client/ |
| D | fcp_runner.cc | 81 const TensorflowSpec& tensorflow_spec, in ConstructOutputsWithDeterministicOrder() argument
85 for (const auto& output_tensor_spec : tensorflow_spec.output_tensor_specs()) { in ConstructOutputsWithDeterministicOrder()
185 client_plan.phase().tensorflow_spec(), in RunPlanWithTensorflowSpec()
202 client_plan.phase().tensorflow_spec(), client_plan.tflite_graph(), in RunPlanWithTensorflowSpec()
|
| D | fl_runner.cc | 133 const TensorflowSpec* tensorflow_spec, in CreateComputationResults() argument
140 if (tensorflow_spec != nullptr) { in CreateComputationResults()
179 tensorflow_spec->output_tensor_specs()) { in CreateComputationResults()
320 return plan_engine.RunPlan(client_plan.phase().tensorflow_spec(), in RunEligibilityEvalPlanWithTensorflowSpec()
334 client_plan.phase().tensorflow_spec(), client_plan.graph(), in RunEligibilityEvalPlanWithTensorflowSpec()
413 const TensorflowSpec& tensorflow_spec, in ConstructOutputsWithDeterministicOrder() argument
418 for (const auto& output_tensor_spec : tensorflow_spec.output_tensor_specs()) { in ConstructOutputsWithDeterministicOrder()
451 client_plan.phase().tensorflow_spec(), in RunPlanWithTensorflowSpec()
467 client_plan.phase().tensorflow_spec(), client_plan.tflite_graph(), in RunPlanWithTensorflowSpec()
484 client_plan.phase().tensorflow_spec(), client_plan.graph(), in RunPlanWithTensorflowSpec()
[all …]
|
| D | lc_runner.cc | 232 client_plan.phase().tensorflow_spec(), client_plan.tflite_graph(), in RunPlanWithTensorflowSpec()
255 client_plan.phase().tensorflow_spec(), client_plan.graph(), in RunPlanWithTensorflowSpec()
|
| /external/federated-compute/fcp/demo/ |
| D | server_test.py | 99 tensorflow_spec=plan_pb2.TensorflowSpec(
|
| /external/federated-compute/fcp/protos/ |
| D | plan.proto | 385 TensorflowSpec tensorflow_spec = 4 [lazy = true]; field
|