diff options
author | 2016-08-26 09:18:21 -0800 | |
---|---|---|
committer | 2016-08-26 10:31:54 -0700 | |
commit | 19e8d26dda94bc4dc0439ca5fddaed8fba4adf1b (patch) | |
tree | 50f172b1363744a1b7529ca21ec4769d61850093 /tensorflow/contrib/quantization | |
parent | d36dd2fcde3e5a95ade152b85268f686173866c5 (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.in | 1 | ||||
-rw-r--r-- | tensorflow/contrib/quantization/kernels/hexagon/quantized_matmul_op_for_hexagon_test.cc | 8 |
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 |