Searched refs:pytree_def (Results 1 – 1 of 1) sorted by relevance
200 xla::PyTreeDef pytree_def; in ParseArguments() local201 pytree_def.FlattenInto(args[i], arguments.flat_dynamic_args); in ParseArguments()202 arguments.signature.dynamic_positional_args_treedef.push_back(pytree_def); in ParseArguments()