diff options
| author |  A. Unique TensorFlower <gardener@tensorflow.org> | 2018-07-02 15:58:54 -0700 |
|---|---|---|
| committer | TensorFlower Gardener <gardener@tensorflow.org> | 2018-07-02 16:02:26 -0700 |
| commit | 0a75d72566b98ee0f91cdf4cb29efb918f46f4da (patch) | |
| tree | 2f7c53aeb1de4c50b5d8c73dd36aad8eb990f5f4 /tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc | |
| parent | 6a948cb2f1ea06270b7fd223ba4c8501b831f838 (diff) | |
Remove reduntant checks in gather op.
PiperOrigin-RevId: 203027634
Diffstat (limited to 'tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc')
| -rw-r--r-- | tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc | 3 |
1 files changed, 0 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc index 82b3ab96fe..8eb0423283 100644 --- a/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc +++ b/tensorflow/contrib/lite/toco/graph_transformations/propagate_fixed_sizes.cc @@ -1039,9 +1039,6 @@ void ProcessGatherOperator(Model* model, GatherOperator* op) { QCHECK_GE(input_shape.dimensions_count(), 1); op->input_rank = input_shape.dimensions_count(); - // We only support 1-D indices. - QCHECK_EQ(indices_shape.dimensions_count(), 1); - // Copy the input dimensions to the output except for dimension 0, // where the dimension of indices_shape is used. // TODO(mgubin): if axis != 0 this is not true, change when it's supported. |