Home
last modified time | relevance | path

Searched refs:outs (Results 1 – 3 of 3) sorted by relevance

/hardware/qcom/neuralnetworks/hvxservice/1.0/
DHexagonOperationsCheck.cpp34 bool addMul(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, in addMul() argument
37 HEXAGON_SOFT_ASSERT_EQ(1, outs.size(), "Need 1 output for " << toString(op)); in addMul()
42 Shape outShape = model->getShape(outs[0]); in addMul()
44 HEXAGON_SOFT_ASSERT(model->setShape(outs[0], outShape), "Error setting shape"); in addMul()
49 bool add(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, HexagonModel* model) { in add() argument
50 return addMul(ins, outs, model, OperationType::ADD); in add()
53 bool mul(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, HexagonModel* model) { in mul() argument
54 return addMul(ins, outs, model, OperationType::MUL); in mul()
57 bool pool(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, HexagonModel* model, in pool() argument
104 Shape outShape = model->getShape(outs[0]); in pool()
[all …]
DHexagonOperationsPrepare.cpp35 bool add(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, HexagonModel* model) { in add() argument
37 HEXAGON_SOFT_ASSERT_EQ(1, outs.size(), "Need 1 output for float32::add"); in add()
46 return model->addFusedFloatOperation(OP_Add_f, NN_PAD_NA, {}, act, {in1, in2}, outs); in add()
49 bool average_pool_2d(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, in average_pool_2d() argument
53 HEXAGON_SOFT_ASSERT_EQ(1, outs.size(), "Need 1 output for float32::average_pool_2d"); in average_pool_2d()
97 outs); in average_pool_2d()
100 bool concatenation(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, in concatenation() argument
103 HEXAGON_SOFT_ASSERT_EQ(1, outs.size(), "Need 1 output for float32::concatenation"); in concatenation()
119 return model->addBasicOperation(OP_Concat_f, NN_PAD_NA, inputs, outs); in concatenation()
122 bool conv_2d(const std::vector<uint32_t>& ins, const std::vector<uint32_t>& outs, in conv_2d() argument
[all …]
DHexagonModel.cpp325 std::vector<hexagon_nn_output> outs = getHexagonOutputs(outputs); in addBasicOperation() local
326 uint32_t node = addOperationInternal(op, pad, inputs, outs); in addBasicOperation()
356 std::vector<hexagon_nn_output> outs = getHexagonOutputs(outputs); in addFloatOperationWithActivation() local
359 uint32_t node = addOperationInternal(op, pad, inputs, outs); in addFloatOperationWithActivation()
364 node = addOperationInternal(activation, NN_PAD_NA, buffer_in, outs); in addFloatOperationWithActivation()
374 std::vector<hexagon_nn_output> outs = getHexagonOutputs(outputs); in addQuant8OperationWithActivation() local
377 uint32_t node = addOperationInternal(op, pad, inputs, outs); in addQuant8OperationWithActivation()
384 node = addOperationInternal(activation, NN_PAD_NA, buffer_in, outs); in addQuant8OperationWithActivation()
395 std::vector<hexagon_nn_output> outs = getHexagonOutputs(outputs); in addFusedFloatOperation() local
399 node = addOperationInternal(op, pad, inputs, outs); in addFusedFloatOperation()
[all …]