diff options
author | Scott Tseng <scott.tseng@kikatech.com> | 2018-04-26 17:30:08 +0800 |
---|---|---|
committer | Scott Tseng <scott.tseng@kikatech.com> | 2018-04-26 17:32:26 +0800 |
commit | afa4748488c55ba5d9dbc9b6c1586132d28d1759 (patch) | |
tree | 1b809a86fcc3c0f5c96c177e5417b1b6e104ae0c /tensorflow/contrib/lite/kernels/topk_v2_test.cc | |
parent | 968addadfd4e4f5688eedc31f92a9066329ff6a7 (diff) |
Fix some issues in official tf.nn.topk() in lite
Diffstat (limited to 'tensorflow/contrib/lite/kernels/topk_v2_test.cc')
-rw-r--r-- | tensorflow/contrib/lite/kernels/topk_v2_test.cc | 2 |
1 files changed, 1 insertions, 1 deletions
diff --git a/tensorflow/contrib/lite/kernels/topk_v2_test.cc b/tensorflow/contrib/lite/kernels/topk_v2_test.cc index 29f2a057cd..212f8acc76 100644 --- a/tensorflow/contrib/lite/kernels/topk_v2_test.cc +++ b/tensorflow/contrib/lite/kernels/topk_v2_test.cc @@ -31,8 +31,8 @@ class TopKV2OpModel : public SingleOpModel { int top_k) { input_ = AddInput(input_type); top_k_ = AddInput(TensorType_INT32); - output_indexes_ = AddOutput(TensorType_INT32); output_values_ = AddOutput(input_type); + output_indexes_ = AddOutput(TensorType_INT32); SetBuiltinOp(BuiltinOperator_TOPK_V2, BuiltinOptions_TopKV2Options, 0); BuildInterpreter({input_shape, {1}}); PopulateTensor<int32_t>(top_k_, {top_k}); |