Searched refs:inputShapes (Results 1 – 5 of 5) sorted by relevance
/frameworks/ml/nn/common/operations/ |
D | Concatenation.cpp | 26 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/ |
D | OperationsUtils.cpp | 371 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 …]
|
D | CpuExecutor.cpp | 1051 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/ |
D | Operations.h | 165 const std::vector<Shape>& inputShapes, int32_t axis, 168 const std::vector<Shape>& inputShapes, int32_t axis,
|
D | OperationsUtils.h | 209 bool concatenationPrepare(const std::vector<Shape>& inputShapes,
|