Remove TensorFlow from doc stings

openmm · Apr 13, 2020 · 3f12cae · 3f12cae
1 parent 29808ca
commit 3f12cae
Show file tree

Hide file tree

Showing 3 changed files with 1 addition and 5 deletions.
diff --git a/openmmapi/include/TensorRTForce.h b/openmmapi/include/TensorRTForce.h
@@ -27,7 +27,7 @@ class OPENMM_EXPORT_NN TensorRTForce : public Force {
     const std::string& getSerializedGraph() const { return serializedGraph; }
     /**
      * Set whether this force makes use of periodic boundary conditions.  If this is set
-     * to true, the TensorFlow graph must include a 3x3 tensor called "boxvectors", which
+     * to true, the TensorRT graph must include a 3x3 tensor called "boxvectors", which
      * is set to the current periodic box vectors.
      */
     void setUsesPeriodicBoundaryConditions(bool periodic) { usePeriodic = periodic; }

diff --git a/openmmapi/include/TensorRTKernels.h b/openmmapi/include/TensorRTKernels.h
@@ -25,8 +25,6 @@ class CalcTesorRTForceKernel : public KernelImpl {
      * 
      * @param system         the System this kernel will be applied to
      * @param force          the TenorRTForce this kernel will be used for
-     * @param session        the TensorFlow session in which to do calculations
-     * @param graph          the TensorFlow graph to use for computing forces and energy
      */
     virtual void initialize(const System& system, const TensorRTForce& force, Engine& engine) = 0;
     /**

diff --git a/platforms/cuda/src/CudaTensorRTKernels.h b/platforms/cuda/src/CudaTensorRTKernels.h
@@ -22,8 +22,6 @@ class CudaCalcTensorRTForceKernel : public CalcTesorRTForceKernel {
      * 
      * @param system         the System this kernel will be applied to
      * @param force          the TensorRTForce this kernel will be used for
-     * @param session        the TensorFlow session in which to do calculations
-     * @param graph          the TensorFlow graph to use for computing forces and energy
      */
     void initialize(const System& system, const TensorRTForce& force, Engine& engine);
     /**