( name: string, op: SimpleUnaryOperation<I, O>, dtype?: DataTypeFor<O>)
| 32 | * kernels that return bool type, such as isFinite, isInf, etc. |
| 33 | */ |
| 34 | export function unaryKernelFunc<I extends number | string = number, |
| 35 | O extends number | string = number>( |
| 36 | name: string, op: SimpleUnaryOperation<I, O>, |
| 37 | dtype?: DataTypeFor<O>): KernelFunc { |
| 38 | |
| 39 | const impl = createSimpleUnaryImpl<I, O>(op); |
| 40 | |
| 41 | return unaryKernelFuncFromImpl<I, O>(name, impl, dtype); |
| 42 | } |
| 43 | |
| 44 | /** |
| 45 | * Template that creates a `KernelFunc` for unary ops from the given |
no test coverage detected
searching dependent graphs…