Home
last modified time | relevance | path

Searched defs:getOutputTensor (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/lite/java/src/main/java/org/tensorflow/lite/
DInterpreterImpl.java160 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in InterpreterImpl
DNativeSignatureRunnerWrapper.java53 public TensorImpl getOutputTensor(String outputName) { in getOutputTensor() method in NativeSignatureRunnerWrapper
DInterpreterApi.java493 Tensor getOutputTensor(int outputIndex); in getOutputTensor() method
DNativeInterpreterWrapper.java429 TensorImpl getOutputTensor(int index) { in getOutputTensor() method in NativeInterpreterWrapper
448 TensorImpl getOutputTensor(String outputName, String signatureKey) { in getOutputTensor() method in NativeInterpreterWrapper
/external/tflite-support/tensorflow_lite_support/java/src/java/org/tensorflow/lite/support/model/
DModel.java236 public Tensor getOutputTensor(int outputIndex) { in getOutputTensor() method in Model
/external/pytorch/test/cpp/c10d/
DProcessGroupNCCLTest.cpp338 at::Tensor getOutputTensor() { in getOutputTensor() function in AllgatherBaseNCCLTest
393 at::Tensor getOutputTensor() { in getOutputTensor() function in ReduceScatterBaseNCCLTest