Searched refs:output_tensor_specs (Results 1 – 10 of 10) sorted by relevance
530 for intrinsic_index, (input_tensor_specs, output_tensor_specs) in enumerate(535 max_struct_length = max(max_input_struct_length, len(output_tensor_specs))562 output_tensor_specs[563 min(i, len(output_tensor_specs) - 1)688 output_tensor_specs=server_result_output_tensors,1250 output_tensor_specs = []1313 output_tensor_specs.append(secagg_tensor_spec.experimental_as_proto())1316 for tensor_spec in output_tensor_specs:1385 if output_tensor_specs:1386 tensorflow_spec.output_tensor_specs.extend(output_tensor_specs)
432 output_tensor_specs = get_flattened_tensor_specs(451 tensor_specs = output_tensor_specs[456 tensor_specs.append(output_tensor_specs[output_tensor_spec_index])
98 tensorflow_spec.output_tensor_specs.append(output_tensor_spec)205 plan_proto.phase[0].client_phase.tensorflow_spec.output_tensor_specs.append(235 client_only_plan_proto.phase.tensorflow_spec.output_tensor_specs.append(
76 output_arrays=[item.name for item in spec.output_tensor_specs],
65 tensorflow_spec.output_tensor_specs()) { in ValidateTensorflowSpec()
139 client_only_plan_.phase().tensorflow_spec().output_tensor_specs()) { in InitializeFlTask()
85 for (const auto& output_tensor_spec : tensorflow_spec.output_tensor_specs()) { in ConstructOutputsWithDeterministicOrder()
179 tensorflow_spec->output_tensor_specs()) { in CreateComputationResults()418 for (const auto& output_tensor_spec : tensorflow_spec.output_tensor_specs()) { in ConstructOutputsWithDeterministicOrder()
434 // Note: both `input_tensor_specs` and `output_tensor_specs` are full443 // - all names in `input_tensor_specs`, `output_tensor_specs`, and446 // - `output_tensor_specs` or `target_node_names` must be non-empty, otherwise482 repeated tensorflow.TensorSpecProto output_tensor_specs = 3; field615 // `output_tensor_specs.name` values.695 // `TensorflowSpec.output_tensor_specs` list.
4449 const ::tensorflow::TensorSpecProto& output_tensor_specs(int index) const; in Swap()4452 output_tensor_specs() const; in Swap()13903 inline const ::tensorflow::TensorSpecProto& TensorflowSpec::output_tensor_specs(int index) const { in output_tensor_specs() function13916 TensorflowSpec::output_tensor_specs() const { in output_tensor_specs() function