Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
PaddlePaddle
Paddle
提交
2e6e188a
P
Paddle
项目概览
PaddlePaddle
/
Paddle
大约 1 年 前同步成功
通知
2297
Star
20931
Fork
5422
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
1423
列表
看板
标记
里程碑
合并请求
543
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
P
Paddle
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
1,423
Issue
1,423
列表
看板
标记
里程碑
合并请求
543
合并请求
543
Pages
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
未验证
提交
2e6e188a
编写于
2月 28, 2023
作者:
H
HongyuJia
提交者:
GitHub
2月 28, 2023
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
[Tensor Operants & Prim-Relevant] Tensor API support default value (#50928)
上级
539293e2
变更
6
隐藏空白更改
内联
并排
Showing
6 changed file
with
27 addition
and
22 deletion
+27
-22
paddle/fluid/prim/utils/static/composite_grad_desc_maker.h
paddle/fluid/prim/utils/static/composite_grad_desc_maker.h
+1
-0
paddle/phi/api/include/tensor.h
paddle/phi/api/include/tensor.h
+16
-17
paddle/phi/api/lib/int_array.cc
paddle/phi/api/lib/int_array.cc
+1
-0
paddle/phi/api/lib/scalar.cc
paddle/phi/api/lib/scalar.cc
+1
-0
paddle/phi/common/int_array.h
paddle/phi/common/int_array.h
+5
-3
paddle/phi/common/scalar.h
paddle/phi/common/scalar.h
+3
-2
未找到文件。
paddle/fluid/prim/utils/static/composite_grad_desc_maker.h
浏览文件 @
2e6e188a
...
...
@@ -28,6 +28,7 @@
#include "paddle/fluid/framework/type_defs.h"
#include "paddle/fluid/prim/utils/static/desc_tensor.h"
#include "paddle/fluid/prim/utils/static/static_global_utils.h"
#include "paddle/phi/api/include/tensor.h"
#include "paddle/phi/core/enforce.h"
#include "paddle/phi/core/flags.h"
...
...
paddle/phi/api/include/tensor.h
浏览文件 @
2e6e188a
...
...
@@ -31,8 +31,10 @@ using gpuStream_t = hipStream_t;
#include "paddle/phi/api/include/dll_decl.h"
#include "paddle/phi/common/data_type.h"
#include "paddle/phi/common/int_array.h"
#include "paddle/phi/common/layout.h"
#include "paddle/phi/common/place.h"
#include "paddle/phi/common/scalar.h"
namespace
phi
{
class
DenseTensor
;
...
...
@@ -47,16 +49,6 @@ namespace paddle {
namespace
experimental
{
class
Tensor
;
template
<
typename
T
>
class
ScalarBase
;
using
Scalar
=
paddle
::
experimental
::
ScalarBase
<
Tensor
>
;
template
<
typename
T
>
class
IntArrayBase
;
using
IntArray
=
paddle
::
experimental
::
IntArrayBase
<
Tensor
>
;
class
AbstractAutogradMeta
{
public:
// No AbstractAutogradMeta should be created
...
...
@@ -684,22 +676,29 @@ class PADDLE_API Tensor final {
Tensor
floor
()
const
;
Tensor
gather_nd
(
const
Tensor
&
index
)
const
;
Tensor
log
()
const
;
Tensor
roll
(
const
IntArray
&
shifts
,
const
std
::
vector
<
int64_t
>&
axis
)
const
;
Tensor
roll
(
const
IntArray
&
shifts
=
{},
const
std
::
vector
<
int64_t
>&
axis
=
{})
const
;
Tensor
scatter
(
const
Tensor
&
index
,
const
Tensor
&
updates
,
bool
overwrite
)
const
;
bool
overwrite
=
true
)
const
;
Tensor
scatter_nd_add
(
const
Tensor
&
index
,
const
Tensor
&
updates
)
const
;
Tensor
abs
()
const
;
Tensor
assign
()
const
;
Tensor
elementwise_pow
(
const
Tensor
&
y
)
const
;
Tensor
expand
(
const
IntArray
&
shape
)
const
;
Tensor
matmul
(
const
Tensor
&
y
,
bool
transpose_x
,
bool
transpose_y
)
const
;
Tensor
max
(
const
IntArray
&
axis
,
bool
keepdim
)
const
;
Tensor
matmul
(
const
Tensor
&
y
,
bool
transpose_x
=
false
,
bool
transpose_y
=
false
)
const
;
Tensor
max
(
const
IntArray
&
axis
=
{},
bool
keepdim
=
false
)
const
;
Tensor
maximum
(
const
Tensor
&
y
)
const
;
Tensor
minimum
(
const
Tensor
&
y
)
const
;
Tensor
scale
(
const
Scalar
&
scale
,
float
bias
,
bool
bias_after_scale
)
const
;
Tensor
sum
(
const
IntArray
&
axis
,
DataType
dtype
,
bool
keepdim
)
const
;
Tensor
tile
(
const
IntArray
&
repeat_times
)
const
;
Tensor
scale
(
const
Scalar
&
scale
=
1.0
,
float
bias
=
0.0
,
bool
bias_after_scale
=
true
)
const
;
Tensor
sum
(
const
IntArray
&
axis
=
{},
DataType
dtype
=
DataType
::
UNDEFINED
,
bool
keepdim
=
false
)
const
;
Tensor
tile
(
const
IntArray
&
repeat_times
=
{})
const
;
};
PADDLE_API
Tensor
operator
+
(
const
Scalar
&
x
,
const
Tensor
&
y
);
...
...
paddle/phi/api/lib/int_array.cc
浏览文件 @
2e6e188a
...
...
@@ -14,6 +14,7 @@ limitations under the License. */
#include "paddle/phi/common/int_array.h"
#include "paddle/phi/api/include/tensor.h"
#include "paddle/phi/api/lib/tensor_copy.h"
#include "paddle/phi/common/place.h"
...
...
paddle/phi/api/lib/scalar.cc
浏览文件 @
2e6e188a
...
...
@@ -14,6 +14,7 @@ limitations under the License. */
#include "paddle/phi/common/scalar.h"
#include "paddle/phi/api/include/tensor.h"
#include "paddle/phi/api/lib/tensor_copy.h"
#include "paddle/phi/common/place.h"
#include "paddle/phi/core/enforce.h"
...
...
paddle/phi/common/int_array.h
浏览文件 @
2e6e188a
...
...
@@ -14,8 +14,10 @@ limitations under the License. */
#pragma once
#include <vector>
#include "paddle/phi/api/ext/exception.h"
#include "paddle/phi/
api/include/tensor
.h"
#include "paddle/phi/
common/data_type
.h"
namespace
paddle
{
namespace
experimental
{
...
...
@@ -103,8 +105,8 @@ class IntArrayBase {
bool
is_from_tensor_
{
false
};
};
using
IntArray
=
paddle
::
experimental
::
IntArrayBase
<
paddle
::
experimental
::
Tensor
>
;
class
Tensor
;
using
IntArray
=
paddle
::
experimental
::
IntArrayBase
<
Tensor
>
;
}
// namespace experimental
}
// namespace paddle
...
...
paddle/phi/common/scalar.h
浏览文件 @
2e6e188a
...
...
@@ -18,7 +18,7 @@ limitations under the License. */
#include <limits>
#include "paddle/phi/api/ext/exception.h"
#include "paddle/phi/
api/include/tensor
.h"
#include "paddle/phi/
common/data_type
.h"
namespace
paddle
{
namespace
experimental
{
...
...
@@ -228,7 +228,8 @@ void CopyScalar(const ScalarBase<T1>& src, ScalarBase<T2>* dst) {
dst
->
data_
.
c128
=
src
.
data_
.
c128
;
}
using
Scalar
=
paddle
::
experimental
::
ScalarBase
<
paddle
::
experimental
::
Tensor
>
;
class
Tensor
;
using
Scalar
=
paddle
::
experimental
::
ScalarBase
<
Tensor
>
;
}
// namespace experimental
}
// namespace paddle
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录