Home
last modified time | relevance | path

Searched defs:SetOutputs (Results 1 – 5 of 5) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_test_util.cc481 void CSession::SetOutputs(std::initializer_list<TF_Operation*> outputs) { in SetOutputs() function in CSession
490 void CSession::SetOutputs(const std::vector<TF_Output>& outputs) { in SetOutputs() function in CSession
/aosp_15_r20/external/tensorflow/tensorflow/lite/
H A Dinterpreter.cc170 TfLiteStatus Interpreter::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Interpreter
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dfunctional_ops.cc94 Status SetOutputs(const OpKernel* kernel, OpKernelContext* ctx, in SetOutputs() function
/aosp_15_r20/external/tensorflow/tensorflow/lite/core/
H A Dsubgraph.cc601 TfLiteStatus Subgraph::SetOutputs(std::vector<int> outputs) { in SetOutputs() function in tflite::Subgraph
/aosp_15_r20/packages/modules/NeuralNetworks/tools/test_generator/
Dtest_generator.py446 def SetOutputs(self, outs): member in Operation