XLA 算子支持列表

目录

• 1. 介绍
  • 1.1. 简介
• 2. XLA 算子支持列表
  • 2.1. AfterAll
  • 2.2. AllReduce
  • 2.3. AlltoAll
  • 2.4. BatchNormGrad
  • 2.5. BatchNormInference
  • 2.6. BatchNormTraining
  • 2.7. BitcastConvertType
  • 2.8. Broadcast
  • 2.9. BroadcastInDim
  • 2.10. Call
  • 2.11. Cholesky
  • 2.12. Clamp
  • 2.13. Collapse
  • 2.14. CollectivePermute
  • 2.15. Concatenate
  • 2.16. Conditional
  • 2.17. Conv
  • 2.18. ConvWithGeneralPadding
  • 2.19. ConvertElementType
  • 2.20. CrossReplicaSum
  • 2.21. CustomCall
  • 2.22. Dot
  • 2.23. DotGeneral
  • 2.24. DynamicSlice
  • 2.25. DynamicUpdateSlice
  • 2.26. Element-wise binary arithmetic operations:Add
  • 2.27. Element-wise binary arithmetic operations:Sub
  • 2.28. Element-wise binary arithmetic operations:Mul
  • 2.29. Element-wise binary arithmetic operations:Div
  • 2.30. Element-wise binary arithmetic operations:Rem
  • 2.31. Element-wise binary arithmetic operations:Max
  • 2.32. Element-wise binary arithmetic operations:Min
  • 2.33. Element-wise binary arithmetic operations:LogicalAnd
  • 2.34. Element-wise binary arithmetic operations:LogicalOR
  • 2.35. Element-wise comparison operations:Eq
  • 2.36. Element-wise comparison operations:Ne
  • 2.37. Element-wise comparison operations:Ge
  • 2.38. Element-wise comparison operations:Gt
  • 2.39. Element-wise comparison operations:Le
  • 2.40. Element-wise comparison operations:Lt
  • 2.41. Element-wise unary functions:Abs
  • 2.42. Element-wise unary functions:Ceil
  • 2.43. Element-wise unary functions:Cos
  • 2.44. Element-wise unary functions:Exp
  • 2.45. Element-wise unary functions:Floor
  • 2.46. Element-wise unary functions:IsFinite
  • 2.47. Element-wise unary functions:Log
  • 2.48. Element-wise unary functions:LogicalNot
  • 2.49. Element-wise unary functions:PopulationCount
  • 2.50. Element-wise unary functions:Neg
  • 2.51. Element-wise unary functions:Sign
  • 2.52. Element-wise unary functions:Tanh
  • 2.53. Element-wise unary functions:sin
  • 2.54. Element-wise unary functions:atan2
  • 2.55. Element-wise unary functions:power
  • 2.56. Element-wise unary functions:rsqrt
  • 2.57. Element-wise unary functions:sqrt
  • 2.58. Element-wise unary functions:round
  • 2.59. Element-wise unary functions:shift_left
  • 2.60. Element-wise unary functions:shift_right_arithmetic
  • 2.61. Element-wise unary functions:shift_right_logical
  • 2.62. Fft
  • 2.63. Gather
  • 2.64. GetDimensionSize
  • 2.65. GetTupleElement
  • 2.66. Infeed
  • 2.67. Outfeed
  • 2.68. lota
  • 2.69. Map
  • 2.70. Pad
  • 2.71. Recv
  • 2.72. Reduce
  • 2.73. ReducePrecision
  • 2.74. ReduceWindow
  • 2.75. ReplicaId
  • 2.76. Reshape
  • 2.77. Rev
  • 2.78. RngNormal
  • 2.79. RngUniform
  • 2.80. Scatter
  • 2.81. Select
  • 2.82. SelectAndScatter
  • 2.83. Send
  • 2.84. Slice
  • 2.85. Sort
  • 2.86. Transpose
  • 2.87. TriangularSolve
  • 2.88. Tuple
  • 2.89. While
  • 2.90. copy
  • 2.91. Clz
• 3. 附录