Searched refs:_composite_to_tensors (Results 1 – 1 of 1) sorted by relevance
228 def _composite_to_tensors(value, is_batched=False): function296 loop_fn_output_tensors = nest.map_structure(_composite_to_tensors,374 nest.map_structure(_composite_to_tensors, loop_fn_outputs))551 functools.partial(_composite_to_tensors,