Searched refs:_flatten_tensors (Results 1 – 2 of 2) sorted by relevance
31 def _flatten_tensors(tensors): function279 input_tensors, shape = _flatten_tensors(input_tensors)464 input_tensors, shape = _flatten_tensors(input_tensors)576 input_tensors, shape = _flatten_tensors(input_tensors)724 input_tensors, shape = _flatten_tensors(input_tensors)814 input_tensors, shape = _flatten_tensors(input_tensors)
45 ar._flatten_tensors([x, x])