Skip to content
体验新版
项目
组织
正在加载...
登录
切换导航
打开侧边栏
wux_labs
Tensorflow
提交
60ed7ceb
T
Tensorflow
项目概览
wux_labs
/
Tensorflow
通知
1
Star
0
Fork
0
代码
文件
提交
分支
Tags
贡献者
分支图
Diff
Issue
0
列表
看板
标记
里程碑
合并请求
0
DevOps
流水线
流水线任务
计划
Wiki
0
Wiki
分析
仓库
DevOps
项目成员
Pages
T
Tensorflow
项目概览
项目概览
详情
发布
仓库
仓库
文件
提交
分支
标签
贡献者
分支图
比较
Issue
0
Issue
0
列表
看板
标记
里程碑
合并请求
0
合并请求
0
Pages
DevOps
DevOps
流水线
流水线任务
计划
分析
分析
仓库分析
DevOps
Wiki
0
Wiki
成员
成员
收起侧边栏
关闭侧边栏
动态
分支图
创建新Issue
流水线任务
提交
Issue看板
体验新版 GitCode,发现更多精彩内容 >>
提交
60ed7ceb
编写于
8月 09, 2022
作者:
L
Laura Pak
提交者:
TensorFlow Release Automation
8月 26, 2022
浏览文件
操作
浏览文件
下载
电子邮件补丁
差异文件
Re-enable testTensorListReserveWithNonScalarNumElements to work with mlir as well.
PiperOrigin-RevId: 466460987
上级
23cb0d30
变更
2
显示空白变更内容
内联
并排
Showing
2 changed file
with
17 addition
and
0 deletion
+17
-0
tensorflow/core/kernels/list_kernels.cc
tensorflow/core/kernels/list_kernels.cc
+7
-0
tensorflow/python/kernel_tests/data_structures/list_ops_test.py
...flow/python/kernel_tests/data_structures/list_ops_test.py
+10
-0
未找到文件。
tensorflow/core/kernels/list_kernels.cc
浏览文件 @
60ed7ceb
...
...
@@ -31,9 +31,11 @@ limitations under the License.
#include "tensorflow/core/framework/allocator.h"
#include "tensorflow/core/framework/op_kernel.h"
#include "tensorflow/core/framework/register_types.h"
#include "tensorflow/core/framework/tensor_shape.h"
#include "tensorflow/core/framework/tensor_types.h"
#include "tensorflow/core/framework/variant.h"
#include "tensorflow/core/framework/variant_op_registry.h"
#include "tensorflow/core/platform/errors.h"
namespace
tensorflow
{
...
...
@@ -322,6 +324,11 @@ class TensorListReserve : public OpKernel {
void
Compute
(
OpKernelContext
*
c
)
override
{
PartialTensorShape
element_shape
;
OP_REQUIRES_OK
(
c
,
TensorShapeFromTensor
(
c
->
input
(
0
),
&
element_shape
));
OP_REQUIRES
(
c
,
TensorShapeUtils
::
IsScalar
(
c
->
input
(
1
).
shape
()),
errors
::
InvalidArgument
(
"The num_elements to reserve must be a tensor size 1, but got "
,
c
->
input
(
1
).
shape
()));
int32_t
num_elements
=
c
->
input
(
1
).
scalar
<
int32
>
()();
OP_REQUIRES
(
c
,
num_elements
>=
0
,
errors
::
InvalidArgument
(
"The num_elements to reserve must be a "
...
...
tensorflow/python/kernel_tests/data_structures/list_ops_test.py
浏览文件 @
60ed7ceb
...
...
@@ -94,6 +94,16 @@ class ListOpsTest(test_util.TensorFlowTestCase, parameterized.TestCase):
l
=
list_ops
.
tensor_list_pop_back
(
l
,
element_dtype
=
dtypes
.
float32
)
self
.
evaluate
(
l
)
def
testTensorListReserveWithNonScalarNumElements
(
self
):
# list_kernels.cc in tf/core/kernels raises InvalidArgumentError, and
# tf_ops_n_z.cc in tf/compiler/mlir/tf/ir raises UnknownError.
with
self
.
assertRaises
((
errors
.
InvalidArgumentError
,
errors
.
UnknownError
)):
l
=
list_ops
.
tensor_list_reserve
(
element_dtype
=
dtypes
.
float32
,
element_shape
=
[
2
,
3
],
num_elements
=
constant_op
.
constant
([
1
,
1
]))
self
.
evaluate
(
l
)
def
testPopUninitializedTensorUseListElementShape
(
self
):
l
=
list_ops
.
tensor_list_reserve
(
element_dtype
=
dtypes
.
float32
,
element_shape
=
[
2
,
3
],
num_elements
=
3
)
...
...
编辑
预览
Markdown
is supported
0%
请重试
或
添加新附件
.
添加附件
取消
You are about to add
0
people
to the discussion. Proceed with caution.
先完成此消息的编辑!
取消
想要评论请
注册
或
登录