Uses of Class
org.tribuo.util.onnx.ONNXPlaceholder

Packages that use ONNXPlaceholder
Package
Description
org.tribuo.util.onnx
Interfaces and utilities for writing ONNX models from Java.

  • Uses of ONNXPlaceholder in org.tribuo.util.onnx

    Methods in org.tribuo.util.onnx that return ONNXPlaceholder
    Modifier and Type
    Method
    Description
    ONNXPlaceholder
    ONNXContext.floatInput(int featureDimension)
    Creates an input node for this ONNXContext, with the name "input", of dimension [batch_size, featureDimension], and of type float32.
    ONNXPlaceholder
    ONNXContext.floatInput(String name, int featureDimension)
    Creates an input node for this ONNXContext, with the given name, of dimension [batch_size, featureDimension], and of type float32.
    ONNXPlaceholder
    ONNXContext.floatOutput(int outputDimension)
    Creates an output node for this ONNXContext, with the name "output", of dimension [batch_size, outputDimension], and of type float32.
    ONNXPlaceholder
    ONNXContext.floatOutput(String name, int outputDimension)
    Creates an output node for this ONNXContext, with the given name, of dimension [batch_size, outputDimension], and of type float32.