Tensorflow Session.Run在Android设备上运行时给出空的输出张量，但在Ubuntu上运行时工作正常

Question

I have a pre trained Tensorflow protobuf that I load from C++ API and build using Bazel. I get expected output tensors when I run the generated executable on Ubuntu. But when I compile the exact same code for Android using Bazel, I get empty output tensors.

Code I use to get the output (same code used in both Ubuntu and Android):

auto statusPred = session->Run(run_options, input, vNames, {}, &answer, &run_metadata);printLog(answer.size());

The answer has size 0 on Android, but has the right expected size when I run the executable, also built using Bazel on UBuntu machine.

Am I missing something fundamental here? Is there a difference between the way computation is done on Ubuntu and Android because of which the output tensor is empty?

Answer 1

Fixed it by adding "cwise_op_floor.cc", to file tensorflow/core/kernels/BUILD, under filegroup android_extended_ops_group1. I see that the latest version has this Op included under the same filegroup though: https://github.com/tensorflow/tensorflow/blob/master/tensorflow/core/kernels/BUILD#L987