27 const std::vector<const TensorInfo*>& inputs,
28 const std::vector<const TensorInfo*>& outputs,
41 TosaSerializationOperator* op =
new TosaSerializationOperator(Op_UNKNOWN, Attribute_NONE,
nullptr, {}, {});
42 return new TosaSerializationBasicBlock(
"", {op}, {}, {}, {});
49 std::vector<const TensorInfo*> inputs;
52 inputs.push_back(&inputSlot.GetConnection()->GetTensorInfo());
55 std::vector<const TensorInfo*> outputs;
58 outputs.push_back(&outputSlot.GetTensorInfo());
TosaSerializationBasicBlock * GetTosaMappingFromLayer(Layer *layer, bool isMain=false)
Copyright (c) 2021 ARM Limited and Contributors.
TosaSerializationBasicBlock * GetTosaMapping(const LayerType type, const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, const BaseDescriptor &, bool isMain=false)
const std::vector< InputSlot > & GetInputSlots() const
Base class for all descriptors.
TosaSerializationBasicBlock * ConvertAdditionToTosaOperator(const std::vector< const TensorInfo *> &inputs, const std::vector< const TensorInfo *> &outputs, bool isMain)
LayerType GetType() const override
Returns the armnn::LayerType of this layer.
const std::vector< OutputSlot > & GetOutputSlots() const
virtual const BaseDescriptor & GetParameters() const override
If the layer has a descriptor return it.
void SetBasicBlockConstantTensorData(Layer *layer, TosaSerializationBasicBlock *)
LayerType
When adding a new layer, adapt also the LastLayer enum value in the enum class LayerType below...