(pfor_input, _, op_func)
| 1784 | @RegisterPForWithArgs("LowerBound", gen_array_ops.lower_bound) |
| 1785 | @RegisterPForWithArgs("UpperBound", gen_array_ops.upper_bound) |
| 1786 | def _convert_searchsorted(pfor_input, _, op_func): |
| 1787 | pfor_input.stack_inputs() |
| 1788 | sorted_inputs = _flatten_first_two_dims(pfor_input.stacked_input(0)) |
| 1789 | values = _flatten_first_two_dims(pfor_input.stacked_input(1)) |
| 1790 | out_type = pfor_input.get_attr("out_type") |
| 1791 | output = op_func(sorted_inputs, values, out_type) |
| 1792 | return wrap(_unflatten_first_dim( |
| 1793 | output, pfor_input.pfor.loop_len_vector), True) |
| 1794 | |
| 1795 | |
| 1796 | @RegisterPFor("MatrixBandPart") |
nothing calls this directly
no test coverage detected