Home
last modified time | relevance | path

Searched refs:input_tensor_specs (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/federated-compute/fcp/artifact_building/
H A Dfederated_compute_plan_builder.py530 for intrinsic_index, (input_tensor_specs, output_tensor_specs) in enumerate(
534 max_input_struct_length = max([len(x) for x in input_tensor_specs])
538 for j, _ in enumerate(input_tensor_specs):
540 tensor_spec = input_tensor_specs[j][
541 min(i, len(input_tensor_specs[j]) - 1)
677 input_tensor_specs=[
684 input_tensor_specs=[
1381 tensorflow_spec.input_tensor_specs.extend(
H A Dplan_utils_test.py97 tensorflow_spec.input_tensor_specs.append(input_tensor_spec)
202 plan_proto.phase[0].client_phase.tensorflow_spec.input_tensor_specs.append(
232 client_only_plan_proto.phase.tensorflow_spec.input_tensor_specs.append(
H A Dplan_utils.py69 for input_tensor in spec.input_tensor_specs:
/aosp_15_r20/external/federated-compute/fcp/client/engine/
H A Dcommon.cc48 tensorflow_spec.input_tensor_specs()) { in ValidateTensorflowSpec()
/aosp_15_r20/external/federated-compute/fcp/demo/
H A Dserver_test.py101 input_tensor_specs=[
/aosp_15_r20/external/federated-compute/fcp/protos/
H A Dplan.proto434 // Note: both `input_tensor_specs` and `output_tensor_specs` are full
443 // - all names in `input_tensor_specs`, `output_tensor_specs`, and
471 repeated tensorflow.TensorSpecProto input_tensor_specs = 2; field
497 // input_tensor_specs.
686 // `TensorflowSpec.input_tensor_specs` list.
/aosp_15_r20/out/soong/.intermediates/external/federated-compute/federated-compute-cc-proto-lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/fcp/protos/
Dplan.pb.h4431 const ::tensorflow::TensorSpecProto& input_tensor_specs(int index) const; in Swap()
4434 input_tensor_specs() const; in Swap()
13866 inline const ::tensorflow::TensorSpecProto& TensorflowSpec::input_tensor_specs(int index) const { in input_tensor_specs() function
13879 TensorflowSpec::input_tensor_specs() const { in input_tensor_specs() function