Method Compile

tensorflow/compiler/tf2xla/kernels/xla_svd_op.cc:41–47

    }
  }
  void Compile(XlaOpKernelContext* ctx) override {
    auto result = xla::SVD(ctx->Input(0), max_iter_, epsilon_,
                           precision_config_.operand_precision(0));
    ctx->SetOutput(0, result.d);
    ctx->SetOutput(1, result.u);
    ctx->SetOutput(2, result.v);
  }

 private:
  int32 max_iter_;