Home
last modified time | relevance | path

Searched refs:SetCustomInputOutput (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/
H A Dwriter_lib_test.cc175 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{2}, /*outputs=*/{3}, in TEST_P()
227 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{2}, /*outputs=*/{3}, in TEST_P()
231 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0, 1}, /*outputs=*/{4}, in TEST_P()
235 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0, 1}, /*outputs=*/{3}, in TEST_P()
270 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0}, /*outputs=*/{2}, in TEST_P()
H A Dwriter_lib.h73 TfLiteStatus SetCustomInputOutput(int subgraph_index,
142 TfLiteStatus SetCustomInputOutput(const std::vector<int>& inputs,
H A Dwriter_lib.cc413 TfLiteStatus SubgraphWriter::SetCustomInputOutput( in SetCustomInputOutput() function in tflite::SubgraphWriter
502 TfLiteStatus ModelWriter::SetCustomInputOutput( in SetCustomInputOutput() function in tflite::ModelWriter
505 return subgraph_writers_[subgraph_index].SetCustomInputOutput(inputs, outputs, in SetCustomInputOutput()