Searched refs:_stack_and_transpose (Results 1 – 1 of 1) sorted by relevance
805 def _stack_and_transpose(tensor_array): function819 outputs_dict = {output_key: _stack_and_transpose(output)824 [_stack_and_transpose(state_element)