Searched refs:_inputs_with_flattening (Results 1 – 1 of 1) sorted by relevance
1397 def _inputs_with_flattening(pfor_input, input_indices): function1424 inputs = _inputs_with_flattening(pfor_input, dims)1504 inputs = _inputs_with_flattening(pfor_input, [0])1571 inputs = _inputs_with_flattening(pfor_input, flatten_dims)2268 cond_rank > 1, lambda: _inputs_with_flattening(pfor_input, [0, 1, 2]),