Package org.apache.wayang.tensorflow.model

Class Convertor

java.lang.Object
org.apache.wayang.tensorflow.model.Convertor
public class Convertor extends Object

  • Constructor Summary

    Constructors
    Constructor
    Description
    Convertor()
     

  • Method Summary

    Modifier and Type
    Method
    Description
    static org.tensorflow.framework.optimizers.Optimizer
    convert(org.tensorflow.Graph graph, Optimizer optimizer)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, BatchNorm2D op, org.tensorflow.Operand<?> input, org.tensorflow.Operand<org.tensorflow.types.TBool> trainingMode)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, BatchNorm3D op, org.tensorflow.Operand<?> input, org.tensorflow.Operand<org.tensorflow.types.TBool> trainingMode)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, Op op, org.tensorflow.Operand<?>... inputs)
     
    static org.tensorflow.Operand<org.tensorflow.types.TInt32>
    convert(org.tensorflow.op.Ops tf, ArgMax op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Cast op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<org.tensorflow.types.TBool>
    convert(org.tensorflow.op.Ops tf, Eq op, org.tensorflow.Operand<?> left, org.tensorflow.Operand<?> right)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Get op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Input op)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Mean op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Conv2D op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Conv3D op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, ConvLSTM2D op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, CrossEntropyLoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, MSELoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, ReLU op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Sigmoid op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Softmax op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Reshape op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Slice op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, Transpose op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Operand<?>
    convert(org.tensorflow.op.Ops tf, ZeroLike op, org.tensorflow.Operand<?> input)
     
    static org.tensorflow.Tensor
    ndArrayToTensor(org.tensorflow.ndarray.NdArray<?> array)
     

    Methods inherited from class java.lang.Object

    clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

  • Constructor Details

    • Convertor

      public Convertor()

  • Method Details

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, Op op, org.tensorflow.Operand<?>... inputs)

    • convert

      public static org.tensorflow.Operand<org.tensorflow.types.TInt32> convert(org.tensorflow.op.Ops tf, ArgMax op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, BatchNorm2D op, org.tensorflow.Operand<?> input, org.tensorflow.Operand<org.tensorflow.types.TBool> trainingMode)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.Graph graph, org.tensorflow.op.Ops tf, BatchNorm3D op, org.tensorflow.Operand<?> input, org.tensorflow.Operand<org.tensorflow.types.TBool> trainingMode)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Cast op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, ConvLSTM2D op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Conv2D op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Conv3D op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, CrossEntropyLoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, MSELoss op, org.tensorflow.Operand<?> predicted, org.tensorflow.Operand<?> labels)

    • convert

      public static org.tensorflow.Operand<org.tensorflow.types.TBool> convert(org.tensorflow.op.Ops tf, Eq op, org.tensorflow.Operand<?> left, org.tensorflow.Operand<?> right)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Get op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Input op)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Linear op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Mean op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, ReLU op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Reshape op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Sigmoid op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Slice op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Softmax op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, Transpose op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.Operand<?> convert(org.tensorflow.op.Ops tf, ZeroLike op, org.tensorflow.Operand<?> input)

    • convert

      public static org.tensorflow.framework.optimizers.Optimizer convert(org.tensorflow.Graph graph, Optimizer optimizer)

    • ndArrayToTensor

      public static org.tensorflow.Tensor ndArrayToTensor(org.tensorflow.ndarray.NdArray<?> array)