diff options
author | 2018-07-08 19:42:24 -0700 | |
---|---|---|
committer | 2018-07-08 19:47:45 -0700 | |
commit | ff83809afa9062e77809d2b65ffbaee3c0045241 (patch) | |
tree | 580ff678c025ceb10450c1b9391e944b0d65b82d /tensorflow/contrib/lite/kernels/BUILD | |
parent | c1ce428a4499ad38916dd512f56a364859c9edd3 (diff) |
Refactoring ArgMax implementation in preparation for ArgMin.
PiperOrigin-RevId: 203698572
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 6 |
1 files changed, 3 insertions, 3 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index 5500f10e63..e749edc5ee 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -136,7 +136,7 @@ cc_library( srcs = [ "activations.cc", "add.cc", - "arg_max.cc", + "arg_min_max.cc", "audio_spectrogram.cc", "basic_rnn.cc", "batch_to_space_nd.cc", @@ -296,9 +296,9 @@ tf_cc_test( ) tf_cc_test( - name = "arg_max_test", + name = "arg_min_max_test", size = "small", - srcs = ["arg_max_test.cc"], + srcs = ["arg_min_max_test.cc"], tags = [ "tflite_not_portable_ios", ], |