diff options
| author |  Eugene Brevdo <ebrevdo@google.com> | 2018-10-10 08:36:36 -0700 |
|---|---|---|
| committer |  TensorFlower Gardener <gardener@tensorflow.org> | 2018-10-10 08:40:03 -0700 |
| commit | 79af30d357fbe0869e163e1d9dce0cb869b3724f (patch) | |
| tree | aa4789c0aa0e10321afe4d3d84eae5fd0e84af3a /tensorflow/contrib/lite/toco/import_tensorflow.cc | |
| parent | 131f6f8429ffa0511a3d5a6a595843d3d96ec942 (diff) | |
[Grappler] Add RemoveStackStridedSliceSameAxis optimizer.
// Replace operations of the form:
// x = stack((a_0, a_1, ..., a_{n-1}), axis=k)[:,...,i,...]
// with
// a_i
// when the strided slice index `i` is applied in the k'th axis.
//
// Similarly, replace operations of the form:
// x = stack((a_0, a_1, ..., a_{n-1}), axis=k)[:,...,i:i+1,...]
// with
// expand_dims(a_i, axis=k)
//
PiperOrigin-RevId: 216535346
Diffstat (limited to 'tensorflow/contrib/lite/toco/import_tensorflow.cc')
0 files changed, 0 insertions, 0 deletions