Method whileLoop

tensorflow/java/src/main/java/org/tensorflow/Graph.java:319–347

Builds a while loop. @param inputs the loop inputs @param cgBuilder WhileSubgraphBuilder to build the conditional subgraph @param bgBuilder WhileSubgraphBuilder to build the body subgraph @param name name for the loop @return list of loop outputs, of the same length as inputs

(
      Output<?>[] inputs,
      WhileSubgraphBuilder cgBuilder,
      WhileSubgraphBuilder bgBuilder,
      String name)

   * @return list of loop outputs, of the same length as {@code inputs}
   */
  public Output<?>[] whileLoop(
      Output<?>[] inputs,
      WhileSubgraphBuilder cgBuilder,
      WhileSubgraphBuilder bgBuilder,
      String name) {
    int ninputs = inputs.length;
    long[] inputHandles = new long[ninputs];
    int[] inputIndices = new int[ninputs];
    Output<?>[] outputs = new Output<?>[ninputs];

    synchronized (nativeHandleLock) {
      try (Reference ref = ref()) {

        for (int i = 0; i < ninputs; i++) {
          inputHandles[i] = inputs[i].getUnsafeNativeHandle();
          inputIndices[i] = inputs[i].index();
        }

        long[] outputHandlesAndIndices =
            whileLoop(nativeHandle, inputHandles, inputIndices, name, cgBuilder, bgBuilder);

        for (int i = 0, j = ninputs; i < ninputs; ++i, ++j) {
          Operation op = new GraphOperation(this, outputHandlesAndIndices[i]);
          outputs[i] = op.output((int) outputHandlesAndIndices[j]);
        }
      }
      return outputs;
    }
  }

  private final Object nativeHandleLock = new Object();
  private long nativeHandle;