Searched refs:unnest_if_single_tensor (Results 1 – 2 of 2) sorted by relevance
265 layer_inputs = unnest_if_single_tensor(layer_inputs)296 def unnest_if_single_tensor(input_tensors): function
1243 base_layer_utils.unnest_if_single_tensor(input_tensors))