public static interface Graph.WhileSubgraphBuilder
To be used when calling Graph.whileLoop(Output[],
org.tensorflow.Graph.WhileSubgraphBuilder, org.tensorflow.Graph.WhileSubgraphBuilder, String)
Example usage (prior to Java 8):
WhileSubgraphBuilder bodyGraphBuilder = new WhileSubgraphBuilder() { @Override public
void buildSubgraph(Graph bodyGraph, Output<?>[] bodyInputs, Output<?>[] bodyOutputs) { // build
body subgraph } };
Example usage (after Java 8):
WhileSubgraphBuilder bodyGraphBuilder = (bodyGraph, bodyInputs, bodyOutputs) -> { //
build body subgraph };
Modifier and Type | Method and Description |
---|---|
void |
buildSubgraph(Graph g,
Output<?>[] inputs,
Output<?>[] outputs)
To be overridden by user with code to build conditional or body subgraph for a while loop
|
Copyright © 2022. All rights reserved.