(pfor_input, op_type, *args, **kw_args)
| 2544 | @RegisterPForWithArgs("RandomStandardNormal") |
| 2545 | @RegisterPForWithArgs("TruncatedNormal") |
| 2546 | def _convert_random(pfor_input, op_type, *args, **kw_args): |
| 2547 | del args |
| 2548 | del kw_args |
| 2549 | inputs = [pfor_input.unstacked_input(i) for i in range(pfor_input.num_inputs)] |
| 2550 | # inputs[0] is "shape" |
| 2551 | inputs[0] = array_ops.concat( |
| 2552 | [pfor_input.pfor.loop_len_vector, inputs[0]], axis=0) |
| 2553 | logging.warning( |
| 2554 | "Note that %s inside pfor op may not give same output as " |
| 2555 | "inside a sequential loop.", op_type) |
| 2556 | outputs = _create_op( |
| 2557 | op_type, |
| 2558 | inputs, [x.dtype for x in pfor_input.outputs], |
| 2559 | attrs=pfor_input.op.node_def.attr).outputs |
| 2560 | return [wrap(x, True) for x in outputs] |
| 2561 | |
| 2562 | |
| 2563 | @RegisterPFor("RandomGamma") |
nothing calls this directly
no test coverage detected