orbax icon indicating copy to clipboard operation
orbax copied to clipboard

obm: Use original argument names for TF input signatures.

Open copybara-service[bot] opened this issue 1 month ago • 0 comments

obm: Use original argument names for TF input signatures.

Previously, the original input names were substituted as "input_0", "input_1", etc. This change updates the logic to extract input argument names directly from the tf.ConcreteFunction's function_def.signature.input_arg, providing more meaningful names in the OBM representation.

Output names are trickier, as they have different names in FunctionDef and SignatureDef. For now they continue to be generated as "output_0", "output_1", etc.

copybara-service[bot] avatar Nov 07 '25 23:11 copybara-service[bot]