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

Update the api_def

Browse Source 
Signed-off-by: Yong Tang <yong.tang.github@outlook.com>
This commit is contained in:
Yong Tang 2018-03-13 22:04:19 +00:00
parent d2744e5792
commit c9d794ab93
2 changed files with 15 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
tensorflow/core/api_def/base_api/api_def_ExtractGlimpse.pbtxt
View File

@ -48,6 +48,14 @@ END
description: <<END description: <<END
indicates if the noise should be generated using a indicates if the noise should be generated using a
uniform distribution or a Gaussian distribution. uniform distribution or a Gaussian distribution.
END
}
attr {
name: "noise"
description: <<END
indicates if the noise should `uniform`, `gaussian`, or
`zero`. The default is `` which means the the noise type
will be decided by `uniform_noise`.
END END
} }
summary: "Extracts a glimpse from the input tensor." summary: "Extracts a glimpse from the input tensor."

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

@ -183,6 +183,13 @@ TEST(ImageOpsTest, ExtractGlimpse_ShapeFn) {
op.input_tensors.resize(2); op.input_tensors.resize(2);
// Inputs are input, size, offsets. // Inputs are input, size, offsets.
TF_ASSERT_OK(NodeDefBuilder("test", "ExtractGlimpse")
.Input({"input", 0, DT_FLOAT})
.Input({"size", 1, DT_INT32})
.Input({"offsets", 2, DT_FLOAT})
.Attr("uniform_noise", true)
.Attr("noise", "")
.Finalize(&op.node_def));
// Rank and size checks. // Rank and size checks.
INFER_ERROR("Shape must be rank 4 but is rank 5", op, "[1,2,3,4,5];?;?"); INFER_ERROR("Shape must be rank 4 but is rank 5", op, "[1,2,3,4,5];?;?");