absl::MakeSpan(outputs), UseFunction());
  ASSERT_EQ(errors::OK, status_.code()) << status_.message();
  ASSERT_NO_FATAL_FAILURE(CheckTensorValue(outputs[0], {0.0f}, /*dims*/ {},
                                           /*abs_error*/ 0));
  outputs[0]->Unref();
}

TEST_P(CppGradients, TestSparseSoftmaxCrossEntropyWithLogitsGrad) {
  if (UseFunction()) {
    // TODO(b/168850692): Enable this.

class Promise<T extends @Nullable Object> implements IThenable<T> {

  @JsFunction
  interface PromiseExecutorCallbackFn<T extends @Nullable Object> {
    @JsFunction
    interface ResolveCallbackFn<T extends @Nullable Object> {
      void onInvoke(T value);
    }

    @JsFunction
    interface RejectCallbackFn {
      void onInvoke(Object error);
    }

                      /*inputs=*/{x.get()},
                      /*outputs=*/{},
                      /*use_function=*/UseFunction());
  ASSERT_EQ(errors::OK, s.code()) << s.message();
}

TEST_P(UnifiedAPI, TestUnknownShapeTracing) {
  if (!UseFunction()) {
    GTEST_SKIP() << "Tracing only test.";
  }
  if (UseMlir()) {
    // TODO(b/173074167): Remove this.

  }

  AbstractContextPtr immediate_execution_ctx_;
  GradientRegistry registry_;
  Status status_;

 public:
  bool UseMlir() const { return strcmp(std::get<0>(GetParam()), "mlir") == 0; }
  bool UseFunction() const { return std::get<2>(GetParam()); }
};

TEST_P(CppGradients, TestIdentityNGrad) {
  // This test is interesting because the current implementation of GradientTape

  }

  @Test
  @TestMetadata("function.kt")
  public void testFunction() {
    runTest("analysis/analysis-api/testData/components/typeInfoProvider/functionClassKind/function.kt");
  }

  @Test
  @TestMetadata("kFunction.kt")
  public void testKFunction() {

  }

  public void testAsFunction_simplistic() {
    String canonical = "a";
    String not = new String("a");

    Function<String, String> internerFunction =
        Interners.asFunction(Interners.<String>newStrongInterner());

    assertSame(canonical, internerFunction.apply(canonical));
    assertSame(canonical, internerFunction.apply(not));
  }

  public void testNullPointerExceptions() {

  }

  public void testAsFunction_simplistic() {
    String canonical = "a";
    String not = new String("a");

    Function<String, String> internerFunction =
        Interners.asFunction(Interners.<String>newStrongInterner());

    assertSame(canonical, internerFunction.apply(canonical));
    assertSame(canonical, internerFunction.apply(not));
  }

  public void testNullPointerExceptions() {

  public void testForLoopVariable() {
    runTest("analysis/analysis-api/testData/symbols/symbolByPsi/forLoopVariable.kt");
  }

  @Test
  @TestMetadata("function.kt")
  public void testFunction() {
    runTest("analysis/analysis-api/testData/symbols/symbolByPsi/function.kt");
  }

  @Test
  @TestMetadata("functionWithTypeAlias.kt")
  public void testFunctionWithTypeAlias() {

  }

  @Test
  @TestMetadata("function_body_with_if.kt")
  public void testFunction_body_with_if() {
    runTest("analysis/analysis-api/testData/components/expressionInfoProvider/isUsedAsExpression/function_body_with_if.kt");
  }

  @Test
  @TestMetadata("function_equal_block_with_if.kt")
  public void testFunction_equal_block_with_if() {

        runTest("analysis/analysis-api/testData/referenceResolve/arrayAccess/withErrors/getFunction.kt");
      }

      @Test
      @TestMetadata("setFunction.kt")
      public void testSetFunction() {
        runTest("analysis/analysis-api/testData/referenceResolve/arrayAccess/withErrors/setFunction.kt");
      }

      @Test
      @TestMetadata("SetOperatorInc.kt")
      public void testSetOperatorInc() {