Searched refs:SetCustomInputOutput (Results 1 – 3 of 3) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/lite/tools/serialization/ |
| H A D | writer_lib_test.cc | 175 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 D | writer_lib.h | 73 TfLiteStatus SetCustomInputOutput(int subgraph_index,
142 TfLiteStatus SetCustomInputOutput(const std::vector<int>& inputs,
|
| H A D | writer_lib.cc | 413 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()
|