|
- #! /usr/bin/python
- # -*- coding: utf-8 -*-
-
- # load nn ops
- from .load_backend import padding_format
- from .load_backend import preprocess_1d_format
- from .load_backend import preprocess_2d_format
- from .load_backend import preprocess_3d_format
- from .load_backend import nchw_to_nhwc
- from .load_backend import nhwc_to_nchw
- from .load_backend import relu
- from .load_backend import relu6
- from .load_backend import leaky_relu
- from .load_backend import softplus
- from .load_backend import tanh
- from .load_backend import sigmoid
- from .load_backend import softmax
- from .load_backend import bias_add
- from .load_backend import conv1d
- from .load_backend import conv2d
- from .load_backend import conv3d
- from .load_backend import lrn
- from .load_backend import moments
- from .load_backend import max_pool
- from .load_backend import avg_pool
- from .load_backend import max_pool3d
- from .load_backend import avg_pool3d
- from .load_backend import pool
- from .load_backend import depthwise_conv2d
- from .load_backend import Conv1d_transpose
- from .load_backend import Conv2d_transpose
- from .load_backend import Conv3d_transpose
- from .load_backend import GroupConv2D
- from .load_backend import BinaryConv2D
- from .load_backend import DorefaConv2D
-
- from .load_backend import ReLU
- from .load_backend import ReLU6
- from .load_backend import LeakyReLU
- from .load_backend import Softplus
- from .load_backend import Tanh
- from .load_backend import Sigmoid
- from .load_backend import Softmax
- from .load_backend import Conv1D
- from .load_backend import Conv2D
- from .load_backend import Conv3D
- from .load_backend import BiasAdd
- from .load_backend import MaxPool1d
- from .load_backend import MaxPool
- from .load_backend import MaxPool3d
- from .load_backend import AvgPool1d
- from .load_backend import AvgPool
- from .load_backend import AvgPool3d
- from .load_backend import Dropout
- from .load_backend import BatchNorm
- from .load_backend import DepthwiseConv2d
- from .load_backend import SeparableConv1D
- from .load_backend import SeparableConv2D
- from .load_backend import AdaptiveMeanPool1D
- from .load_backend import AdaptiveMeanPool2D
- from .load_backend import AdaptiveMeanPool3D
- from .load_backend import AdaptiveMaxPool1D
- from .load_backend import AdaptiveMaxPool2D
- from .load_backend import AdaptiveMaxPool3D
- from .load_backend import Floor
- from .load_backend import Ceil
-
- # load ops
- from .load_backend import Variable
- from .load_backend import matmul
- from .load_backend import add
- from .load_backend import dtypes
- from .load_backend import minimum
- from .load_backend import reshape
- from .load_backend import concat
- from .load_backend import convert_to_tensor
- from .load_backend import convert_to_numpy
- from .load_backend import sqrt
- from .load_backend import reduce_mean
- from .load_backend import reduce_min
- from .load_backend import reduce_max
- from .load_backend import pad
- from .load_backend import stack
- from .load_backend import meshgrid
- from .load_backend import range
- from .load_backend import expand_dims
- from .load_backend import tile
- from .load_backend import cast
- from .load_backend import transpose
- from .load_backend import gather_nd
- from .load_backend import clip_by_value
- from .load_backend import split
- from .load_backend import get_tensor_shape
- from .load_backend import set_context
- from .load_backend import resize
- from .load_backend import floor
- from .load_backend import gather
- from .load_backend import linspace
- from .load_backend import slice
- from .load_backend import add_n
- from .load_backend import ceil
- from .load_backend import multiply
- from .load_backend import divide
- from .load_backend import identity
-
- # dtype
- from .load_backend import (DType, float16, float32, float64, int8, int16, int32, int64, uint8, uint16, uint32, uint64)
- # initlizers
- from .load_backend import (zeros, ones, constant, random_uniform, random_normal, truncated_normal, he_normal)
- # backend
- from .load_backend import BACKEND
- from .load_backend import BACKEND_VERSION
-
- from .load_backend import Reshape
- from .load_backend import ReduceSum
- from .load_backend import ReduceMax
- from .load_backend import ReduceMean
- from .load_backend import OneHot
- from .load_backend import L2Normalize
- from .load_backend import EmbeddingLookup
- from .load_backend import NCELoss
- from .load_backend import NotEqual
- from .load_backend import Cast
- from .load_backend import ExpandDims
- from .load_backend import CountNonzero
- from .load_backend import FlattenReshape
- from .load_backend import Transpose
- from .load_backend import MatMul
- from .load_backend import Tile
- from .load_backend import Concat
- from .load_backend import ZeroPadding1D
- from .load_backend import ZeroPadding2D
- from .load_backend import ZeroPadding3D
- from .load_backend import Stack
- from .load_backend import Unstack
- from .load_backend import Sign
- from .load_backend import Resize
- from .load_backend import Pad
- from .load_backend import Minimum
- from .load_backend import Maximum
- from .load_backend import Meshgrid
- from .load_backend import BatchToSpace
- from .load_backend import DepthToSpace
|