zoo.pipeline.api.onnx.mapper package

Submodules

zoo.pipeline.api.onnx.mapper.abs module

class zoo.pipeline.api.onnx.mapper.abs.AbsMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.add module

class zoo.pipeline.api.onnx.mapper.add.AddMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.averagepool module

class zoo.pipeline.api.onnx.mapper.averagepool.AveragePoolMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.batchnormalization module

class zoo.pipeline.api.onnx.mapper.batchnormalization.BatchNormalizationMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

to_zoo_format(trainable_values)

Convert ONNX _initializer to Zoo format :return: list of ndarray

zoo.pipeline.api.onnx.mapper.cast module

class zoo.pipeline.api.onnx.mapper.cast.CastMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.clip module

class zoo.pipeline.api.onnx.mapper.clip.ClipMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.concat module

class zoo.pipeline.api.onnx.mapper.concat.ConcatMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.constant module

class zoo.pipeline.api.onnx.mapper.constant.ConstantMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.conv module

class zoo.pipeline.api.onnx.mapper.conv.ConvMapper(node, initializer, inputs)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

to_zoo_format(trainable_values)

Convert ONNX _initializer to Zoo format :return: list of ndarray

zoo.pipeline.api.onnx.mapper.div module

class zoo.pipeline.api.onnx.mapper.div.DivMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.dropout module

class zoo.pipeline.api.onnx.mapper.dropout.DropoutMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.elu module

class zoo.pipeline.api.onnx.mapper.elu.EluMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.exp module

class zoo.pipeline.api.onnx.mapper.exp.ExpMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.flatten module

class zoo.pipeline.api.onnx.mapper.flatten.FlattenMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.gather module

class zoo.pipeline.api.onnx.mapper.gather.GatherMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.gemm module

class zoo.pipeline.api.onnx.mapper.gemm.GemmMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

to_zoo_format(trainable_values)

Convert ONNX _initializer to Zoo format :return: list of ndarray

zoo.pipeline.api.onnx.mapper.globalaveragepool module

class zoo.pipeline.api.onnx.mapper.globalaveragepool.GlobalAveragePoolMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.greater module

class zoo.pipeline.api.onnx.mapper.greater.GreaterMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.hardsigmoid module

class zoo.pipeline.api.onnx.mapper.hardsigmoid.HardSigmoidMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.leakyrelu module

class zoo.pipeline.api.onnx.mapper.leakyrelu.LeakyReluMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.log module

class zoo.pipeline.api.onnx.mapper.log.LogMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.logsoftmax module

class zoo.pipeline.api.onnx.mapper.logsoftmax.LogSoftmaxMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.lrn module

class zoo.pipeline.api.onnx.mapper.lrn.LRNMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.matmul module

class zoo.pipeline.api.onnx.mapper.matmul.MatMulMapper(node, initializer, inputs)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.maxpool module

class zoo.pipeline.api.onnx.mapper.maxpool.MaxPoolMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.mul module

class zoo.pipeline.api.onnx.mapper.mul.MulMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.neg module

class zoo.pipeline.api.onnx.mapper.neg.NegMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.operator_mapper module

class zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper(node, initializer, inputs)

Bases: object

static of(node, _params, inputs)

to_tensor()

Convert a node to tensor

to_zoo_format(trainable_values)

Convert ONNX _initializer to Zoo format :return: list of ndarray

zoo.pipeline.api.onnx.mapper.pow module

class zoo.pipeline.api.onnx.mapper.pow.PowMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.reducemean module

class zoo.pipeline.api.onnx.mapper.reducemean.ReduceMeanMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.reducesum module

class zoo.pipeline.api.onnx.mapper.reducesum.ReduceSumMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.relu module

class zoo.pipeline.api.onnx.mapper.relu.ReluMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.reshape module

class zoo.pipeline.api.onnx.mapper.reshape.ReshapeMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.shape module

class zoo.pipeline.api.onnx.mapper.shape.ShapeMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.sigmoid module

class zoo.pipeline.api.onnx.mapper.sigmoid.SigmoidMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.slice module

class zoo.pipeline.api.onnx.mapper.slice.SliceMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.softmax module

class zoo.pipeline.api.onnx.mapper.softmax.SoftmaxMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.sqrt module

class zoo.pipeline.api.onnx.mapper.sqrt.SqrtMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.squeeze module

class zoo.pipeline.api.onnx.mapper.squeeze.SqueezeMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.sub module

class zoo.pipeline.api.onnx.mapper.sub.SubMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.tanh module

class zoo.pipeline.api.onnx.mapper.tanh.TanhMapper(node, _params, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.transpose module

class zoo.pipeline.api.onnx.mapper.transpose.TransposeMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

zoo.pipeline.api.onnx.mapper.unsqueeze module

class zoo.pipeline.api.onnx.mapper.unsqueeze.UnsqueezeMapper(node, initializer, _all_tensors)

Bases: zoo.pipeline.api.onnx.mapper.operator_mapper.OperatorMapper

Module contents