blob: d003d529cc316dfde63f76284f98ae698e1d8034 (
plain)
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
|
/* Copyright 2018 The TensorFlow Authors. All Rights Reserved.
Licensed under the Apache License, Version 2.0 (the "License");
you may not use this file except in compliance with the License.
You may obtain a copy of the License at
http://www.apache.org/licenses/LICENSE-2.0
Unless required by applicable law or agreed to in writing, software
distributed under the License is distributed on an "AS IS" BASIS,
WITHOUT WARRANTIES OR CONDITIONS OF ANY KIND, either express or implied.
See the License for the specific language governing permissions and
limitations under the License.
==============================================================================*/
#ifndef TENSORFLOW_COMPILER_XLA_CLIENT_LIB_MATH_H_
#define TENSORFLOW_COMPILER_XLA_CLIENT_LIB_MATH_H_
#include "tensorflow/compiler/xla/client/xla_client/xla_builder.h"
namespace xla {
// Computes the square root of 'operand'.
XlaOp Sqrt(XlaOp operand);
// Computes the reciprocal of the square root of 'operand'.
XlaOp Rsqrt(XlaOp operand);
// Computes the square of 'operand'.
XlaOp Square(XlaOp operand);
// Computes the reciprocal of 'operand'.
XlaOp Reciprocal(XlaOp operand);
// Evaluates a polynomial given coefficients and `x`.
// N.B. Coefficients should be supplied in decreasing order.
XlaOp EvaluatePolynomial(XlaOp x,
tensorflow::gtl::ArraySlice<float> coefficients);
// Computes an approximation of the error function complement (1 - erf(x)).
XlaOp Erfc(XlaOp x);
// Computes an approximation of the error function.
XlaOp Erf(XlaOp x);
// Computes an approximation of the inverse of the error function.
XlaOp ErfInv(XlaOp x);
// Computes an approximation of the lgamma function.
XlaOp Lgamma(XlaOp input);
// Computes an approximation of the digamma function.
XlaOp Digamma(XlaOp input);
// Trigonometric functions
// Computes the arc cosine of 'x'.
XlaOp Acos(XlaOp x);
// Computes the arc sine of 'x'.
XlaOp Asin(XlaOp x);
// Computes the arc tangent of 'x'.
XlaOp Atan(XlaOp x);
// Computes the tangent of 'x'.
XlaOp Tan(XlaOp x);
// Hyperbolic trigonometric functions
// Computes the inverse hyperbolic cosine of 'x'.
XlaOp Acosh(XlaOp x);
// Computes the inverse hyperbolic sine of 'x'.
XlaOp Asinh(XlaOp x);
// Computes the inverse hyperbolic tangent of 'x'.
XlaOp Atanh(XlaOp x);
// Computes the hyperbolic cosine of 'x'.
XlaOp Cosh(XlaOp x);
// Computes the hyperbolic sine of 'x'.
XlaOp Sinh(XlaOp x);
} // namespace xla
#endif // TENSORFLOW_COMPILER_XLA_CLIENT_LIB_MATH_H_
|
