(pfor_input, op_type, *args, **kw_args)
| 2484 | @RegisterPForWithArgs("RsqrtGrad") |
| 2485 | @RegisterPForWithArgs("ReciprocalGrad") |
| 2486 | def _convert_grads(pfor_input, op_type, *args, **kw_args): |
| 2487 | del args |
| 2488 | del kw_args |
| 2489 | # TODO(agarwal): Looks like these ops don't support broadcasting. Hence we |
| 2490 | # have to use tiling here. |
| 2491 | pfor_input.stack_inputs() |
| 2492 | outputs = _create_op( |
| 2493 | op_type, [x.t for x in pfor_input.inputs], |
| 2494 | [x.dtype for x in pfor_input.outputs], |
| 2495 | attrs=pfor_input.op.node_def.attr).outputs |
| 2496 | return [wrap(x, True) for x in outputs] |
| 2497 | |
| 2498 | |
| 2499 | @RegisterPFor("Select") |
nothing calls this directly
no test coverage detected