Function createSimpleUnaryImpl

tfjs-backend-cpu/src/utils/unary_impl.ts:25–36

(op: SimpleUnaryOperation<I, O>)

 * Template that creates implementation for unary op.
 */
export function createSimpleUnaryImpl<I extends number | string = number,
  O extends number | string = number>(op: SimpleUnaryOperation<I, O>):
    SimpleUnaryImpl<I, O> {
  return (values, dtype, attrs) => {
    const newValues =
        util.getArrayFromDType(dtype, values.length);
    for (let i = 0; i < values.length; ++i) {
      newValues[i] = op(values[i], attrs);
    }
    return newValues;
  };
}