You can not select more than 25 topics Topics must start with a chinese character,a letter or number, can include dashes ('-') and can be up to 35 characters long.

__init__.py 5.1 kB

4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
4 years ago
123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657585960616263646566676869707172737475767778798081828384858687888990919293949596979899100101102103104105106107108109110111112113114115116117118119120121122123124125126127128129130131132133134135136137138139140141142143144145146147
  1. #! /usr/bin/python
  2. # -*- coding: utf-8 -*-
  3. # load nn ops
  4. from .load_backend import padding_format
  5. from .load_backend import preprocess_1d_format
  6. from .load_backend import preprocess_2d_format
  7. from .load_backend import preprocess_3d_format
  8. from .load_backend import nchw_to_nhwc
  9. from .load_backend import nhwc_to_nchw
  10. from .load_backend import relu
  11. from .load_backend import relu6
  12. from .load_backend import leaky_relu
  13. from .load_backend import softplus
  14. from .load_backend import tanh
  15. from .load_backend import sigmoid
  16. from .load_backend import softmax
  17. from .load_backend import bias_add
  18. from .load_backend import conv1d
  19. from .load_backend import conv2d
  20. from .load_backend import conv3d
  21. from .load_backend import lrn
  22. from .load_backend import moments
  23. from .load_backend import max_pool
  24. from .load_backend import avg_pool
  25. from .load_backend import max_pool3d
  26. from .load_backend import avg_pool3d
  27. from .load_backend import pool
  28. from .load_backend import depthwise_conv2d
  29. from .load_backend import Conv1d_transpose
  30. from .load_backend import Conv2d_transpose
  31. from .load_backend import Conv3d_transpose
  32. from .load_backend import GroupConv2D
  33. from .load_backend import BinaryConv2D
  34. from .load_backend import DorefaConv2D
  35. from .load_backend import ReLU
  36. from .load_backend import ReLU6
  37. from .load_backend import LeakyReLU
  38. from .load_backend import Softplus
  39. from .load_backend import Tanh
  40. from .load_backend import Sigmoid
  41. from .load_backend import Softmax
  42. from .load_backend import Conv1D
  43. from .load_backend import Conv2D
  44. from .load_backend import Conv3D
  45. from .load_backend import BiasAdd
  46. from .load_backend import MaxPool1d
  47. from .load_backend import MaxPool
  48. from .load_backend import MaxPool3d
  49. from .load_backend import AvgPool1d
  50. from .load_backend import AvgPool
  51. from .load_backend import AvgPool3d
  52. from .load_backend import Dropout
  53. from .load_backend import BatchNorm
  54. from .load_backend import DepthwiseConv2d
  55. from .load_backend import SeparableConv1D
  56. from .load_backend import SeparableConv2D
  57. from .load_backend import AdaptiveMeanPool1D
  58. from .load_backend import AdaptiveMeanPool2D
  59. from .load_backend import AdaptiveMeanPool3D
  60. from .load_backend import AdaptiveMaxPool1D
  61. from .load_backend import AdaptiveMaxPool2D
  62. from .load_backend import AdaptiveMaxPool3D
  63. from .load_backend import Floor
  64. from .load_backend import Ceil
  65. # load ops
  66. from .load_backend import Variable
  67. from .load_backend import matmul
  68. from .load_backend import add
  69. from .load_backend import dtypes
  70. from .load_backend import minimum
  71. from .load_backend import reshape
  72. from .load_backend import concat
  73. from .load_backend import convert_to_tensor
  74. from .load_backend import convert_to_numpy
  75. from .load_backend import sqrt
  76. from .load_backend import reduce_mean
  77. from .load_backend import reduce_min
  78. from .load_backend import reduce_max
  79. from .load_backend import pad
  80. from .load_backend import stack
  81. from .load_backend import meshgrid
  82. from .load_backend import range
  83. from .load_backend import expand_dims
  84. from .load_backend import tile
  85. from .load_backend import cast
  86. from .load_backend import transpose
  87. from .load_backend import gather_nd
  88. from .load_backend import clip_by_value
  89. from .load_backend import split
  90. from .load_backend import get_tensor_shape
  91. from .load_backend import set_context
  92. from .load_backend import resize
  93. from .load_backend import floor
  94. from .load_backend import gather
  95. from .load_backend import linspace
  96. from .load_backend import slice
  97. from .load_backend import add_n
  98. from .load_backend import ceil
  99. from .load_backend import multiply
  100. from .load_backend import divide
  101. from .load_backend import identity
  102. # dtype
  103. from .load_backend import (DType, float16, float32, float64, int8, int16, int32, int64, uint8, uint16, uint32, uint64)
  104. # initlizers
  105. from .load_backend import (zeros, ones, constant, random_uniform, random_normal, truncated_normal, he_normal)
  106. # backend
  107. from .load_backend import BACKEND
  108. from .load_backend import BACKEND_VERSION
  109. from .load_backend import Reshape
  110. from .load_backend import ReduceSum
  111. from .load_backend import ReduceMax
  112. from .load_backend import ReduceMean
  113. from .load_backend import OneHot
  114. from .load_backend import L2Normalize
  115. from .load_backend import EmbeddingLookup
  116. from .load_backend import NCELoss
  117. from .load_backend import NotEqual
  118. from .load_backend import Cast
  119. from .load_backend import ExpandDims
  120. from .load_backend import CountNonzero
  121. from .load_backend import FlattenReshape
  122. from .load_backend import Transpose
  123. from .load_backend import MatMul
  124. from .load_backend import Tile
  125. from .load_backend import Concat
  126. from .load_backend import ZeroPadding1D
  127. from .load_backend import ZeroPadding2D
  128. from .load_backend import ZeroPadding3D
  129. from .load_backend import Stack
  130. from .load_backend import Unstack
  131. from .load_backend import Sign
  132. from .load_backend import Resize
  133. from .load_backend import Pad
  134. from .load_backend import Minimum
  135. from .load_backend import Maximum
  136. from .load_backend import Meshgrid
  137. from .load_backend import BatchToSpace
  138. from .load_backend import DepthToSpace
  139. from .load_backend import rnncell
  140. from .load_backend import lstmcell
  141. from .load_backend import grucell
  142. from .load_backend import rnnbase

TensorLayer3.0 是一款兼容多种深度学习框架为计算后端的深度学习库。计划兼容TensorFlow, Pytorch, MindSpore, Paddle.