Searched refs:numInputTensors (Results 1 – 2 of 2) sorted by relevance
105 const size_t numInputTensors = ins.size() - 1; in concatenation() local108 std::vector<hexagon_nn_input> inputs(numInputTensors + 1); in concatenation()109 for (size_t i = 0; i < numInputTensors; ++i) { in concatenation()114 const int32_t axis = model->getScalar<int32_t>(ins[numInputTensors]); in concatenation()562 const size_t numInputTensors = ins.size() - 1; in concatenation() local565 std::vector<hexagon_nn_input> inputs(numInputTensors * 3 + 1); in concatenation()566 for (size_t i = 0; i < numInputTensors; ++i) { in concatenation()567 inputs[i + 1 + numInputTensors * 0] = model->getTensor(ins[i]); in concatenation()568 inputs[i + 1 + numInputTensors * 1] = model->getQuantizationMin(ins[i]); in concatenation()569 inputs[i + 1 + numInputTensors * 2] = model->getQuantizationMax(ins[i]); in concatenation()[all …]
135 const size_t numInputTensors = ins.size() - 1; in concatenation() local137 const int32_t axis = model->getScalar<int32_t>(ins[numInputTensors]); in concatenation()140 std::vector<Shape> inShapes(numInputTensors); in concatenation()141 for (size_t i = 0; i < numInputTensors; ++i) { in concatenation()