Searched refs:_inputs_with_flattening (Results 1 – 1 of 1) sorted by relevance
1741 def _inputs_with_flattening(pfor_input, input_indices): function1775 inputs = _inputs_with_flattening(pfor_input, dims)1902 inputs = _inputs_with_flattening(pfor_input, [0])1970 inputs = _inputs_with_flattening(pfor_input, flatten_dims)3039 cond_rank > 1, lambda: _inputs_with_flattening(pfor_input, [0, 1, 2]),