aboutsummaryrefslogtreecommitdiffhomepage
path: root/tensorflow/contrib/quantization
diff options
context:
space:
mode:
authorGravatar A. Unique TensorFlower <gardener@tensorflow.org>2016-08-26 09:18:21 -0800
committerGravatar TensorFlower Gardener <gardener@tensorflow.org>2016-08-26 10:31:54 -0700
commit19e8d26dda94bc4dc0439ca5fddaed8fba4adf1b (patch)
tree50f172b1363744a1b7529ca21ec4769d61850093 /tensorflow/contrib/quantization
parentd36dd2fcde3e5a95ade152b85268f686173866c5 (diff)
Add a way to link and call hexagon shared library APIs
Change: 131413727
Diffstat (limited to 'tensorflow/contrib/quantization')
-rw-r--r--tensorflow/contrib/quantization/Makefile.in1
-rw-r--r--tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc8
2 files changed, 6 insertions, 3 deletions
diff --git a/tensorflow/contrib/quantization/Makefile.in b/tensorflow/contrib/quantization/Makefile.in
index 656f1d00ea..563639e5d7 100644
--- a/tensorflow/contrib/quantization/Makefile.in
+++ b/tensorflow/contrib/quantization/Makefile.in
@@ -39,6 +39,7 @@ tensorflow/contrib/quantization/kernels/quantize_op.cc \
tensorflow/contrib/quantization/kernels/quantized_conv_ops.cc \
tensorflow/contrib/quantization/kernels/quantized_matmul_op.cc \
tensorflow/contrib/quantization/kernels/quantized_matmul_op_test.cc \
+tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc \
tensorflow/contrib/makefile/test/test_main.cc
QUANTIZATION_TEST_OBJS := $(addprefix $(OBJDIR), $(QUANTIZATION_TEST_SRCS:.cc=.o))
diff --git a/tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc b/tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc
index 1fa4ba0163..9c452fe39c 100644
--- a/tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc
+++ b/tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc
@@ -30,7 +30,7 @@ limitations under the License.
#include "tensorflow/core/platform/test.h"
#ifdef USE_HEXAGON_LIBS
-#include "gemm_wrapper.h"
+#include "tensorflow/core/platform/hexagon/gemm_wrapper.h"
#endif
namespace tensorflow {
@@ -39,8 +39,10 @@ class QuantizedMatMulOpForHexagonTest : public OpsTestBase {
protected:
void SetUp() final {
#ifdef USE_HEXAGON_LIBS
- LOG(INFO) << "Hexagon libs are linked (version = "
- << hexagon_gemm_wrapper_GetVersion() << ")";
+ LOG(INFO) << "Hexagon libs are linked (wrapper version = "
+ << hexagon_gemm_wrapper_GetWrapperVersion()
+ << ", hexagon binary version = "
+ << hexagon_gemm_wrapper_GetHexagonBinaryVersion() << ")";
#else
LOG(WARNING) << "Hexagon libs are not linked.";
#endif