Searched refs:output_flatten (Results 1 – 2 of 2) sorted by relevance
179 output_flatten = input_flatten183 output_flatten = lambda x: nest.flatten(x) if output_is_sequence else [x] function208 dtype_flat = output_flatten(dtype)259 flat_fn_values = output_flatten(packed_fn_values)
369 output_flatten = input_flatten373 output_flatten = lambda x: nest.flatten(x) if output_is_sequence else [x] function419 initializer_flat = output_flatten(initializer)457 flat_a_out = output_flatten(a_out)