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()
|