Function z

tfjs-core/src/kernel_registry_test.ts:318–322

(...x: tf.Tensor[])

    // Inputs as array.
    const z = (...x: tf.Tensor[]) =>
        // tslint:disable-next-line: no-unnecessary-type-assertion
        tf.engine().runKernel(
            kernelName, x as unknown as tf.NamedTensorMap, {} /* attrs */) as
        tf.Tensor;
    const gradFunc = tf.grads(z);
    const dx = gradFunc(x);
    expect(kernelWasCalled).toBe(true);