未验证 提交 b075dee8 编写于 作者: Z zyfncg 提交者: GitHub

[PHI] Support some c++ api in paddle namespace (#41778)

* support some c++ api in paddle namespace

* change c++ api namespace in custom op
上级 ad9585b6
...@@ -24,6 +24,114 @@ limitations under the License. */ ...@@ -24,6 +24,114 @@ limitations under the License. */
namespace paddle { namespace paddle {
using Tensor = experimental::Tensor; using Tensor = experimental::Tensor;
// using several Tensor initialize functions in paddle namespace // using several Tensor initialize functions in paddle namespace
using experimental::abs;
using experimental::acos;
using experimental::acosh;
using experimental::add;
using experimental::allclose;
using experimental::argsort;
using experimental::asin;
using experimental::asinh;
using experimental::atan;
using experimental::atan2;
using experimental::atanh;
using experimental::bernoulli;
using experimental::ceil;
using experimental::cholesky;
using experimental::cholesky_solve;
using experimental::clip;
using experimental::concat;
using experimental::conj;
using experimental::cos;
using experimental::cosh;
using experimental::cross;
using experimental::det;
using experimental::diag;
using experimental::diagonal;
using experimental::digamma;
using experimental::dist;
using experimental::divide;
using experimental::dot;
using experimental::elu;
using experimental::empty; using experimental::empty;
using experimental::empty_like;
using experimental::equal_all;
using experimental::erf;
using experimental::erfinv;
using experimental::exp;
using experimental::expand;
using experimental::expm1;
using experimental::flatten;
using experimental::flip;
using experimental::floor;
using experimental::floor_divide;
using experimental::full; using experimental::full;
using experimental::gather;
using experimental::gather_nd;
using experimental::gelu;
using experimental::gumbel_softmax;
using experimental::imag;
using experimental::increment;
using experimental::index_sample;
using experimental::is_empty;
using experimental::isclose;
using experimental::isfinite;
using experimental::isinf;
using experimental::isnan;
using experimental::kron;
using experimental::kthvalue;
using experimental::label_smooth;
using experimental::lerp;
using experimental::lgamma;
using experimental::log;
using experimental::log10;
using experimental::log1p;
using experimental::log2;
using experimental::logit;
using experimental::masked_select;
using experimental::matmul;
using experimental::matrix_power;
using experimental::maximum;
using experimental::maxout;
using experimental::minimum;
using experimental::mode;
using experimental::multi_dot;
using experimental::multinomial;
using experimental::multiply;
using experimental::mv;
using experimental::nll_loss;
using experimental::one_hot;
using experimental::pixel_shuffle;
using experimental::poisson;
using experimental::qr;
using experimental::real;
using experimental::reciprocal;
using experimental::relu;
using experimental::reshape;
using experimental::roll;
using experimental::round;
using experimental::rsqrt;
using experimental::scatter;
using experimental::scatter_nd_add;
using experimental::selu;
using experimental::sign;
using experimental::silu;
using experimental::sin;
using experimental::sinh;
using experimental::split;
using experimental::sqrt;
using experimental::square;
using experimental::stack;
using experimental::strided_slice;
using experimental::subtract;
using experimental::tanh;
using experimental::thresholded_relu;
using experimental::tile;
using experimental::trace;
using experimental::triangular_solve;
using experimental::unbind;
using experimental::unique;
using experimental::unsqueeze;
using experimental::where;
} // namespace paddle } // namespace paddle
...@@ -20,8 +20,7 @@ limitations under the License. */ ...@@ -20,8 +20,7 @@ limitations under the License. */
std::vector<paddle::Tensor> PhiLinearForward(const paddle::Tensor& x, std::vector<paddle::Tensor> PhiLinearForward(const paddle::Tensor& x,
const paddle::Tensor& weight, const paddle::Tensor& weight,
const paddle::Tensor& bias) { const paddle::Tensor& bias) {
return { return {paddle::add(paddle::matmul(x, weight), bias)};
paddle::experimental::add(paddle::experimental::matmul(x, weight), bias)};
} }
std::vector<std::vector<int64_t>> LinearInferShape( std::vector<std::vector<int64_t>> LinearInferShape(
......
...@@ -2167,7 +2167,7 @@ ...@@ -2167,7 +2167,7 @@
data_type : x data_type : x
- api : unsqueeze - api : unsqueeze
args : (Tensor x, IntArray axes) args : (Tensor x, IntArray axis)
output : Tensor(xshape), Tensor(out) output : Tensor(xshape), Tensor(out)
infer_meta : infer_meta :
func : UnsqueezeInferMeta func : UnsqueezeInferMeta
......
Markdown is supported
0% .
You are about to add 0 people to the discussion. Proceed with caution.
先完成此消息的编辑!
想要评论请 注册