Searched refs:SetCustomInputOutput (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/tools/serialization/ |
| D | writer_lib_test.cc | 169 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{2}, /*outputs=*/{3}, in TEST_P()
221 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{2}, /*outputs=*/{3}, in TEST_P()
225 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0, 1}, /*outputs=*/{4}, in TEST_P()
229 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0, 1}, /*outputs=*/{3}, in TEST_P()
264 EXPECT_EQ(writer.SetCustomInputOutput(/*inputs=*/{0}, /*outputs=*/{2}, in TEST_P()
|
| D | writer_lib.h | 113 TfLiteStatus SetCustomInputOutput(const std::vector<int>& inputs,
|
| D | writer_lib.cc | 386 TfLiteStatus SubgraphWriter::SetCustomInputOutput( in SetCustomInputOutput() function in tflite::SubgraphWriter
|