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.

tensorflow-op-combination.md 2.1 kB

123456789101112131415161718192021222324252627282930313233343536373839404142434445464748495051525354555657
  1. ## batchnorm
  2. ```
  3. Input A 0 1 A 0 0 0
  4. MemoryData sub/y 0 1 sub/y 16 0 0
  5. BinaryOp sub 2 1 A sub/y sub 1
  6. MemoryData div/y 0 1 div/y 16 0 0
  7. BinaryOp div 2 1 sub div/y div 3
  8. MemoryData mul/y 0 1 mul/y 16 0 0
  9. BinaryOp mul 2 1 div mul/y mul 2
  10. MemoryData BiasAdd/bias 0 1 BiasAdd/bias 16 0 0
  11. BinaryOp BiasAdd 2 1 mul BiasAdd/bias BiasAdd 0
  12. ```
  13. ## convolution
  14. ```
  15. Input A 0 1 A 0 0 0
  16. Convolution Conv2D 1 1 A Conv2D 10 3 1 1 0 0 270
  17. MemoryData biases/read 0 1 biases/read 10 0 0
  18. BinaryOp BiasAdd 2 1 Conv2D biases/read BiasAdd 0
  19. ```
  20. ## innerproduct
  21. ```
  22. Input A 0 1 A 0 0 0
  23. MemoryData biases/read 0 1 biases/read 10 0 0
  24. InnerProduct MatMul 1 1 A MatMul 10 0 2560
  25. BinaryOp conv6 2 1 MatMul biases/read conv6 0
  26. ```
  27. ## leakyrelu
  28. ```
  29. Input A 0 1 A 0 0 0
  30. Split splitncnn_0 1 2 A A_splitncnn_0 A_splitncnn_1
  31. MemoryData mul_1/x 0 1 mul_1/x 0 0 0
  32. BinaryOp mul_1 2 1 mul_1/x A_splitncnn_1 mul_1 2
  33. BinaryOp leaky 2 1 mul_1 A_splitncnn_0 leaky 4
  34. ```
  35. ## prelu
  36. ```
  37. Input A 0 1 A 0 0 0
  38. Split splitncnn_0 1 2 A A_splitncnn_0 A_splitncnn_1
  39. MemoryData prelu/alpha 0 1 prelu/alpha 10 0 0
  40. ReLU prelu/Relu 1 1 A_splitncnn_1 prelu/Relu 0.000000
  41. UnaryOp prelu/Neg 1 1 A_splitncnn_0 prelu/Neg 1
  42. ReLU prelu/Relu_1 1 1 prelu/Neg prelu/Relu_1 0.000000
  43. UnaryOp prelu/Neg_1 1 1 prelu/Relu_1 prelu/Neg_1 1
  44. BinaryOp prelu/Mul 2 1 prelu/alpha prelu/Neg_1 prelu/Mul 2
  45. BinaryOp prelu/add 2 1 prelu/Relu prelu/Mul prelu/add 0
  46. ```
  47. ## softmax
  48. ```
  49. Input A 0 1 A 0 0 0
  50. Split splitncnn_4 1 2 A A_splitncnn_0 A_splitncnn_1
  51. Reduction Max 1 1 A_splitncnn_1 Max 4 -2 1.000000
  52. BinaryOp sub 2 1 A_splitncnn_0 Max sub 1
  53. UnaryOp Exp 1 1 sub Exp 7
  54. Split splitncnn_5 1 2 Exp Exp_splitncnn_0 Exp_splitncnn_1
  55. Reduction Sum 1 1 Exp_splitncnn_1 Sum 0 -2 1.000000
  56. BinaryOp prob 2 1 Exp_splitncnn_0 Sum prob 3
  57. ```