Home
last modified time | relevance | path

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

/frameworks/ml/nn/common/operations/
DConcatenation.cpp26 const std::vector<Shape>& inputShapes, int32_t axis, in concatenationFloat32() argument
28 int num_inputs = inputShapes.size(); in concatenationFloat32()
32 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenationFloat32()
45 const std::vector<Shape>& inputShapes, int32_t axis, in concatenationQuant8() argument
47 int num_inputs = inputShapes.size(); in concatenationQuant8()
51 inputDims[i] = convertShapeToDims(inputShapes[i]); in concatenationQuant8()
/frameworks/ml/nn/common/
DOperationsUtils.cpp371 bool concatenationPrepare(const std::vector<Shape>& inputShapes, in concatenationPrepare() argument
375 int num_inputs = inputShapes.size(); in concatenationPrepare()
376 OperandType input_type = inputShapes[0].type; in concatenationPrepare()
377 uint32_t num_dimensions = getNumberOfDimensions(inputShapes[0]); in concatenationPrepare()
382 int sumAxis = getSizeOfDimension(inputShapes[0], axis); in concatenationPrepare()
384 NN_OPS_CHECK(getNumberOfDimensions(inputShapes[i]) == num_dimensions); in concatenationPrepare()
385 NN_OPS_CHECK(inputShapes[i].type == inputShapes[0].type); in concatenationPrepare()
387 NN_OPS_CHECK(inputShapes[0].offset == inputShapes[i].offset); in concatenationPrepare()
388 NN_OPS_CHECK(inputShapes[0].scale == inputShapes[i].scale); in concatenationPrepare()
392 sumAxis += getSizeOfDimension(inputShapes[i], axis); in concatenationPrepare()
[all …]
DCpuExecutor.cpp1051 std::vector<Shape> inputShapes(numInputTensors); in executeOperation() local
1056 inputShapes[i] = input.shape(); in executeOperation()
1059 success = concatenationPrepare(inputShapes, axis, &outShape) && in executeOperation()
1061 concatenationFloat32(inputDataPtrs, inputShapes, axis, in executeOperation()
1064 std::vector<Shape> inputShapes(numInputTensors); in executeOperation() local
1069 inputShapes[i] = input.shape(); in executeOperation()
1072 success = concatenationPrepare(inputShapes, axis, &outShape) && in executeOperation()
1074 concatenationQuant8(inputDataPtrs, inputShapes, axis, in executeOperation()
/frameworks/ml/nn/common/include/
DOperations.h165 const std::vector<Shape>& inputShapes, int32_t axis,
168 const std::vector<Shape>& inputShapes, int32_t axis,
DOperationsUtils.h209 bool concatenationPrepare(const std::vector<Shape>& inputShapes,