Home
last modified time | relevance | path

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

/aosp12/packages/modules/NeuralNetworks/common/operations/
H A DConcatenation.cpp54 int num_inputs = inputShapes.size(); in concatenation()
58 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenation()
74 int num_inputs = inputShapes.size();
80 inputScales[i] = inputShapes[i].scale;
81 inputOffsets[i] = inputShapes[i].offset;
82 inputDims[i] = convertShapeToDims(inputShapes[i]);
99 std::vector<Shape> inputShapes; in concatenation() local
104 inputShapes.push_back(context->getInputShape(i)); in concatenation()
123 std::vector<Shape> inputShapes; variable
126 inputShapes.push_back(context->getInputShape(i));
[all …]