Searched defs:pytree_def (Results 1 – 1 of 1) sorted by relevance
248 xla::PyTreeDef& pytree_def = arguments.signature.dynamic_arg_treedefs[i]; in ParseArguments() local260 xla::PyTreeDef& pytree_def = in ParseArguments() local311 xla::PyTreeDef& pytree_def = in ParseArguments() local