experiments/STT-tensorflow
1
0
Fork 0
Code Issues Pull Requests Projects Releases Wiki Activity

Fix //tensorflow/core:ops_sparse_ops_test

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2018-11-04 18:43:45 +00:00
parent 74784ac463
commit e64bfb83dc
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
tensorflow/core/ops/sparse_ops_test.cc
View File

@ -133,6 +133,13 @@ TEST(SparseOpsTest, SparseToDense_ShapeFn) {
TEST(SparseOpsTest, SparseReduceSum_ShapeFn) {
ShapeInferenceTestOp op("SparseReduceSum");
TF_ASSERT_OK(NodeDefBuilder("test", "SparseReduceSum")
.Input({"input_indices", 0, DT_INT64})
.Input({"input_values", 1, DT_INT64})
.Input({"input_shape", 2, DT_INT64})
.Input({"reduction_axes", 3, DT_INT32})
.Attr("keep_dims", false)
.Finalize(&op.node_def));
// Shape fn always yields unknown.
INFER_OK(op, "?;?;?;?", "?");