forked from TensorLayer/tensorlayer3
147 lines
5.0 KiB
Python
147 lines
5.0 KiB
Python
#! /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
|
|
from .load_backend import rnncell
|
|
from .load_backend import lstmcell
|
|
from .load_backend import grucell
|
|
from .load_backend import rnnbase |