diff options
author | 2016-09-09 09:21:29 -0800 | |
---|---|---|
committer | 2016-09-09 10:31:16 -0700 | |
commit | 29faf36ed2341fe50a02a1ad1329b96157d96d55 (patch) | |
tree | 10a8162d5bea048421a98cf4a33c963a0be11aa7 /tensorflow/core/ops/string_ops.cc | |
parent | b7541f67c21b5a12120af0b7ac33404bd5160643 (diff) |
Switch ops in functional_ops, math_ops, state_ops, string_ops,
contrib/quantization, and contrib/layers to use C++ shape inference functions.
Implement Betainc C++ shape inference function; it's a little different from
the python one, mainly because propagating an unknown input (a_shape in this
case) when all are unknown is not correct for C++ (in case we later backwards
bind the value), since it could end up being a broadcasted scalar.
Change ReduceJoin's C++ shape inference function to match python.
Fix SymbolicGradient's shape function in C++ - there are actually more inputs
than outputs.
Changes QuantizedBiasAdd to be more precise (the C++ shape fn invokes the common bias_add to compute the shape of the first output).
Change: 132688147
Diffstat (limited to 'tensorflow/core/ops/string_ops.cc')
-rw-r--r-- | tensorflow/core/ops/string_ops.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/core/ops/string_ops.cc b/tensorflow/core/ops/string_ops.cc index 248fb7cbbb..fac5b20097 100644 --- a/tensorflow/core/ops/string_ops.cc +++ b/tensorflow/core/ops/string_ops.cc @@ -95,7 +95,7 @@ REGISTER_OP("ReduceJoin") .Attr("keep_dims: bool = false") .Attr("separator: string = ''") .Output("output: string") - .SetShapeFn(shape_inference::ReductionShape) + .SetShapeFn(shape_inference::ReductionShapeForReduceJoin) .Doc(R"doc( Joins a string Tensor across the given dimensions. |