Function predict

e2e/benchmarks/benchmark_util_test.js:109–115

()

      it('profile all statements in async predict function', async () => {
        const predict = async () => {
          await sleep(1);
          const x = tf.tensor1d([1, 2, 3]);
          const x2 = x.square();
          x2.dispose();
          return x;
        };

        const oldTensorCount = tf.memory().numTensors;
        let profileInfo = await profileInference(predict);