Skip to content

Commit 11d1940

Browse files
authored
* Re-enable training APIs in presets for ONNX Runtime (pull #1774)
1 parent fd82f54 commit 11d1940

6 files changed

Lines changed: 882 additions & 8 deletions

File tree

onnxruntime/cppbuild.sh

Lines changed: 1 addition & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -14,7 +14,7 @@ export COREML_FLAGS=
1414
export OPENVINO_FLAGS=
1515
export OPENVINO_CMAKE_DIR=
1616
export OPENMP_FLAGS= # "--use_openmp"
17-
export TRAINING_FLAGS= # --enable_training_apis --enable_training_ops
17+
export TRAINING_FLAGS="--enable_training_apis --enable_training_ops"
1818
export CUDAFLAGS="-v"
1919
export CUDACXX="/usr/local/cuda/bin/nvcc"
2020
export CUDA_HOME="/usr/local/cuda"
Lines changed: 23 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -0,0 +1,23 @@
1+
// Targeted by JavaCPP version 1.5.14-SNAPSHOT: DO NOT EDIT THIS FILE
2+
3+
package org.bytedeco.onnxruntime;
4+
5+
import java.nio.*;
6+
import org.bytedeco.javacpp.*;
7+
import org.bytedeco.javacpp.annotation.*;
8+
9+
import static org.bytedeco.javacpp.presets.javacpp.*;
10+
import org.bytedeco.opencl.*;
11+
import static org.bytedeco.opencl.global.OpenCL.*;
12+
import org.bytedeco.dnnl.*;
13+
import static org.bytedeco.dnnl.global.dnnl.*;
14+
15+
import static org.bytedeco.onnxruntime.global.onnxruntime.*;
16+
// Type that enables performing training for the given user models.
17+
@Opaque @Properties(inherit = org.bytedeco.onnxruntime.presets.onnxruntime.class)
18+
public class OrtCheckpointState extends Pointer {
19+
/** Empty constructor. Calls {@code super((Pointer)null)}. */
20+
public OrtCheckpointState() { super((Pointer)null); }
21+
/** Pointer cast constructor. Invokes {@link Pointer#Pointer(Pointer)}. */
22+
public OrtCheckpointState(Pointer p) { super(p); }
23+
}

0 commit comments

Comments
 (0)