projects / platform / upstream / tensorflow.git / commit
? search:
summary | shortlog | log | commit | commitdiff | tree
(parent: b1347d0) | patch
Remove check for axis == 3 since if the input dimension is not 4, the input axis...
authorA. Unique TensorFlower <gardener@tensorflow.org>
Wed, 16 May 2018 04:35:27 +0000 (21:35 -0700)
committerTensorFlower Gardener <gardener@tensorflow.org>
Wed, 16 May 2018 04:38:40 +0000 (21:38 -0700)
commit38cb05291393c002c874902fdc5c55006cc2e78f
tree240c9403c78180ea4aba27c819ca3781d4b65567tree | snapshot
parentb1347d0e3110dd20ce13bc8b6fb67548db6c3d0fcommit | diff
Remove check for axis == 3 since if the input dimension is not 4, the input axis is not necessary 3.
And change the test as well.

PiperOrigin-RevId: 196777020
tensorflow/contrib/lite/kernels/arg_max.cc diff | blob | history
tensorflow/contrib/lite/kernels/internal/optimized/optimized_ops.h diff | blob | history
tensorflow/contrib/lite/kernels/internal/reference/reference_ops.h diff | blob | history
tensorflow/contrib/lite/testing/generate_examples.py diff | blob | history
tensorflow/contrib/lite/testing/generated_examples_zip_test.cc diff | blob | history
Domain: Machine Learning / ML Framework;