ONE - On-device Neural Engine
|
#include "Frontend.h"
#include "Context.h"
#include "Convert.h"
#include "TensorBags.h"
#include "GraphBuilderRegistry.h"
#include <nncc/core/ADT/tensor/LexicalLayout.h>
#include <nncc/core/ADT/tensor/Shape.h>
#include <iostream>
Go to the source code of this file.
Namespaces | |
namespace | tflimport |
Functions | |
void | tflimport::set_module_inputs (coco::Module *m, TensorContext &ctx, TensorBags &bags, const IndexVector &inputs) |
Set module input operands and its information. | |
void | tflimport::set_module_outputs (coco::Module *m, TensorContext &ctx, TensorBags &bags, const IndexVector &outputs) |
Set module output operands and its information. | |
void | tflimport::copy_tensors (GraphBuilderContext *ctx) |
Copy values of tfl tensors into coco::Data if the data was not copied. | |