Commit 20669f62 authored by resec's avatar resec Committed by Jonathan Hseu
Browse files

Add interface for long(int64) datatype (#11632)

Added below methods for TensorFlowInferenceInterface
public void feed(String inputName, long[] src, long... dims)
public void feed(String inputName, LongBuffer src, long... dims)
public void fetch(String outputName, long[] dst)
public void fetch(String outputName, LongBuffer dst)

They are very useful in case when we are handling int64 Tensor (e.g. output of tf.argmax)

I do not see the unit test file for TensorFlowInferenceInterface.java, so I did not add any test case for the change. But this change is very straight forward. Please let me know if I missed any. Thanks.
parent 107e3073
Loading
Loading
Loading
Loading
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Please to comment