Searched refs:SubgraphWriter (Results 1 – 4 of 4) sorted by relevance
81 flatbuffers::Offset<flatbuffers::Vector<T_OUTPUT>> SubgraphWriter::ExportVector( in ExportVector()88 SubgraphWriter::ExportOperators(flatbuffers::FlatBufferBuilder* fbb) { in ExportOperators()158 SubgraphWriter::ExportTensors(flatbuffers::FlatBufferBuilder* fbb) { in ExportTensors()249 SubgraphWriter::ExportBuffers(flatbuffers::FlatBufferBuilder* fbb) { in ExportBuffers()254 SubgraphWriter::CreateOpCodeTable(flatbuffers::FlatBufferBuilder* fbb) { in CreateOpCodeTable()259 std::vector<int> SubgraphWriter::RemapTensorIndicesToWritten(const T& input) { in RemapTensorIndicesToWritten()275 TfLiteStatus SubgraphWriter::GetBuffer(std::unique_ptr<uint8_t[]>* out, in GetBuffer()299 flatbuffers::Offset<SubGraph> SubgraphWriter::PopulateAndGetOffset( in PopulateAndGetOffset()311 TfLiteStatus SubgraphWriter::Write(const std::string& filename) { in Write()318 TfLiteStatus SubgraphWriter::RegisterCustomWriter( in RegisterCustomWriter()[all …]
24 `lite/core/subgraph.h`) you can use `SubgraphWriter`.32 tflite::SubgraphWriter writer(&interpreter->subgraph(subgraph_index));37 `SubgraphWriter` supports custom ops and/or custom I/O tensors.41 Both `ModelWriter` and `SubgraphWriter` support a `GetBuffer` method to return
50 SubgraphWriter writer(&interpreter->primary_subgraph()); in WriteToFile()84 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()94 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()168 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()218 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()263 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()370 SubgraphWriter writer(&interpreter.primary_subgraph()); in TEST_P()
77 class SubgraphWriter {88 explicit SubgraphWriter(Subgraph* subgraph) in SubgraphWriter() function119 explicit SubgraphWriter( in SubgraphWriter() function