(shape)
| 54 | |
| 55 | |
| 56 | def truncated_normal(shape): |
| 57 | return execute( |
| 58 | b'TruncatedNormal', |
| 59 | 1, |
| 60 | inputs=[shape], |
| 61 | attrs=('dtype', dtypes.float32.as_datatype_enum, 'T', |
| 62 | shape.dtype.as_datatype_enum, 'seed', 0, 'seed2', 0))[0] |
| 63 | |
| 64 | |
| 65 | def current_device(): |
no test coverage detected