diff options
author | 2018-09-21 09:24:42 +0800 | |
---|---|---|
committer | 2018-09-21 09:24:42 +0800 | |
commit | a3321411e4f7b5ba3f1773de8a557947a4851671 (patch) | |
tree | c105462de626f7a58b6ef6d6d702cb86a4295618 /tensorflow/core/framework | |
parent | d7a8e852941e8cd856caafddf7c93d857e83e8b1 (diff) |
More changes.
Diffstat (limited to 'tensorflow/core/framework')
-rw-r--r-- | tensorflow/core/framework/common_shape_fns.cc | 5 | ||||
-rw-r--r-- | tensorflow/core/framework/common_shape_fns.h | 2 |
2 files changed, 5 insertions, 2 deletions
diff --git a/tensorflow/core/framework/common_shape_fns.cc b/tensorflow/core/framework/common_shape_fns.cc index 20922d7884..5f34a75121 100644 --- a/tensorflow/core/framework/common_shape_fns.cc +++ b/tensorflow/core/framework/common_shape_fns.cc @@ -1306,6 +1306,8 @@ Status RandomShape(shape_inference::InferenceContext* c) { return Status::OK(); } +namespace { + // This SliceHelper processes the output shape of the `slice` // when the tensor of `sizes` is available. template <typename T> @@ -1333,6 +1335,7 @@ Status SliceHelper(InferenceContext* c, ShapeHandle begin_value, return Status::OK(); } +} // namespace Status SliceShape(InferenceContext* c) { ShapeHandle input = c->input(0); @@ -1356,7 +1359,7 @@ Status SliceShape(InferenceContext* c) { // We check the tensor value here and will only use // `MakeShapeFromShapeTensor` when `sizes_value` is null. - // The reason is that `sizes`might contain -1, which can't + // The reason is that `sizes` might contain -1, which can't // be represented (-1 in the ShapeHandle would mean "unknown"). const Tensor* sizes_value = c->input_tensor(2); diff --git a/tensorflow/core/framework/common_shape_fns.h b/tensorflow/core/framework/common_shape_fns.h index 478f796516..3a496e06ae 100644 --- a/tensorflow/core/framework/common_shape_fns.h +++ b/tensorflow/core/framework/common_shape_fns.h @@ -293,7 +293,7 @@ inline Status BroadcastBinaryOpShapeFn(InferenceContext* c) { // Shape function for random operations. Status RandomShape(shape_inference::InferenceContext* c); -// Shape function for Slice operator. +// Shape function for Slice opertaions. Status SliceShape(shape_inference::InferenceContext* c); // Validates the 3 component tensors of a sparse tensor have the proper |