diff options
author | 2018-04-09 22:01:59 -0700 | |
---|---|---|
committer | 2018-04-09 22:04:16 -0700 | |
commit | 3f3a6e6685449130389e0b8d76f6ba0fc457bcfe (patch) | |
tree | 6634bcc3cc6ad37e999de22a828c5f120b3d7082 /tensorflow/contrib/lite/kernels/BUILD | |
parent | 1c8f3c81698b67b8fffce86c97df27d392b84cb8 (diff) |
Implementation of ArgMax
PiperOrigin-RevId: 192236845
Diffstat (limited to 'tensorflow/contrib/lite/kernels/BUILD')
-rw-r--r-- | tensorflow/contrib/lite/kernels/BUILD | 17 |
1 files changed, 17 insertions, 0 deletions
diff --git a/tensorflow/contrib/lite/kernels/BUILD b/tensorflow/contrib/lite/kernels/BUILD index df0f3cbeb0..b79900623e 100644 --- a/tensorflow/contrib/lite/kernels/BUILD +++ b/tensorflow/contrib/lite/kernels/BUILD @@ -135,6 +135,7 @@ cc_library( srcs = [ "activations.cc", "add.cc", + "arg_max.cc", "audio_spectrogram.cc", "basic_rnn.cc", "batch_to_space_nd.cc", @@ -271,6 +272,22 @@ tf_cc_test( ) tf_cc_test( + name = "arg_max_test", + size = "small", + srcs = ["arg_max_test.cc"], + tags = [ + "tflite_not_portable_ios_arm64", + "tflite_not_portable_ios_x86_64", + ], + deps = [ + ":builtin_ops", + "//tensorflow/contrib/lite:framework", + "//tensorflow/contrib/lite/kernels:test_util", + "@com_google_googletest//:gtest", + ], +) + +tf_cc_test( name = "div_test", size = "small", srcs = ["div_test.cc"], |