projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: 2585a81) | patch
Add logic for StridedSlice ops in ShapeRefiner::ConstantPartialShape().
authorSkye Wanderman-Milne <skyewm@google.com>
Tue, 8 May 2018 00:28:41 +0000 (17:28 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Tue, 8 May 2018 01:32:07 +0000 (18:32 -0700)
commit1af09b57ef663d4ab0c02a00e2af1f1e2819d32f
treef56df608f376a0bf594bb30ae9fc99c1c1d4dfa4tree | snapshot
parent2585a8181904b39c71fc314940587c02b30a68a6commit | diff
Add logic for StridedSlice ops in ShapeRefiner::ConstantPartialShape().

This mimics the logic in tensor_util.constant_value_as_shape, allowing
the C++ shape inference code to infer more shapes than it could before.

This change also adds an optional stride argument to InferenceContext::Subshape().

PiperOrigin-RevId: 195749522
tensorflow/core/common_runtime/shape_refiner.cc diff | blob | history
tensorflow/core/common_runtime/shape_refiner.h diff | blob | history
tensorflow/core/common_runtime/shape_refiner_test.cc diff | blob | history
tensorflow/core/framework/shape_inference.cc diff | blob | history
tensorflow/core/framework/shape_inference.h diff | blob | history
Domain: Machine Learning / ML Framework;