Home
last modified time | relevance | path

Searched refs:SubGraphContext (Results 1 – 15 of 15) sorted by relevance

/packages/modules/NeuralNetworks/runtime/operation_converters/
DSubGraphContext.cpp28 SubGraphContext::SubGraphContext(const Model* model, const Model::Subgraph* subgraph, in SubGraphContext() function in android::nn::SubGraphContext
49 SubGraphFlatbuffer SubGraphContext::finish() { in finish()
54 int SubGraphContext::addTensorFlatbuffer(TensorFlatbuffer tensor, int32_t operandIdx) { in addTensorFlatbuffer()
65 void SubGraphContext::addOperatorFlatbuffer(OperatorFlatbuffer opFlatbuffer) { in addOperatorFlatbuffer()
69 void SubGraphContext::addSubGraphInput(int32_t operandIdx) { in addSubGraphInput()
74 void SubGraphContext::addSubGraphOutput(int32_t operandIdx) { in addSubGraphOutput()
79 uint32_t SubGraphContext::addOpCode(OperationType operationType) { in addOpCode()
104 int SubGraphContext::getTensorIdxFromOperandIdx(int operandIdx) const { in getTensorIdxFromOperandIdx()
108 const Mapping& SubGraphContext::getMapping(uint32_t poolIndex) { in getMapping()
122 std::pair<const uint8_t*, uint32_t> SubGraphContext::getConstantPointerAndLength( in getConstantPointerAndLength()
[all …]
DConv2DOperationConverter.h29 Result<void> convert(const Operation& operation, SubGraphContext* context) const override;
33 SubGraphContext* context) const;
35 SubGraphContext* context) const;
39 SubGraphContext* context) const;
DLogisticOperationConverter.h29 Result<void> convert(const Operation& operation, SubGraphContext* context) const override;
33 SubGraphContext* context) const;
35 SubGraphContext* context) const;
DSubGraphContext.h32 class SubGraphContext {
34 SubGraphContext(const Model* model, const Model::Subgraph* subgraph,
93 void SubGraphContext::copyConstantValueToData(const Operand& operand, Type* data, size_t dataSize) { in copyConstantValueToData()
101 Type SubGraphContext::getConstantScalar(const Operand& operand) { in getConstantScalar()
DLogisticOperationConverter.cpp28 const Operation& operation, SubGraphContext* context) const { in getLogisticInputs()
36 const Operation& operation, SubGraphContext* context) const { in getLogisticOutputs()
44 SubGraphContext* context) const { in convert()
DArithmeticOperationConverter.h30 SubGraphContext* context) const;
32 SubGraphContext* context) const;
DArithmeticOperationConverter.cpp28 const Operation& operation, SubGraphContext* context) const { in getArithmeticInputs()
38 const Operation& operation, SubGraphContext* context) const { in getArithmeticOutputs()
DConv2DOperationConverter.cpp28 const Operation& operation, SubGraphContext* context) const { in getConv2DInputs()
45 const Operation& operation, SubGraphContext* context) const { in getConv2DOutputs()
53 SubGraphContext* context) const { in decomposeExplicitPadding()
144 SubGraphContext* context) const { in convert()
DOperationConverter.h29 virtual Result<void> convert(const Operation& operation, SubGraphContext* context) const = 0;
DAddOperationConverter.h29 Result<void> convert(const Operation& operation, SubGraphContext* context) const override;
DDepthwiseConv2DOperationConverter.h29 Result<void> convert(const Operation& operation, SubGraphContext* context) const override;
DAddOperationConverter.cpp28 SubGraphContext* context) const { in convert()
DDepthwiseConv2DOperationConverter.cpp28 SubGraphContext* context) const { in convert()
/packages/modules/NeuralNetworks/runtime/
DFlatbufferModelBuilder.cpp90 SubGraphContext context(&mModel, &subgraph, &mBuilder, &mOpCodesVector, in createSubGraphFlatbuffer()
DAndroid.bp185 "operation_converters/SubGraphContext.cpp",