Searched refs:outputDataPtrs (Results 1 – 3 of 3) sorted by relevance
| /packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Split.cpp | 32 const std::vector<Scalar*>* outputDataPtrs, in splitGeneric() argument
47 for (size_t i = 0; i < outputDataPtrs->size(); ++i) { in splitGeneric()
49 memcpy(outputDataPtrs->at(i) + k * copySize, inputPtr, copySize * sizeof(Scalar)); in splitGeneric()
58 const std::vector<_Float16*>* outputDataPtrs, in splitFloat16() argument
61 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
65 const std::vector<float*>* outputDataPtrs, in splitFloat32() argument
68 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
72 const std::vector<uint8_t*>* outputDataPtrs, in splitQuant8() argument
75 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
79 const std::vector<int8_t*>* outputDataPtrs, in splitQuant8Signed() argument
[all …]
|
| /packages/modules/NeuralNetworks/common/include/ |
| D | Operations.h | 112 const std::vector<_Float16*>* outputDataPtrs,
116 const std::vector<float*>* outputDataPtrs,
120 const std::vector<int32_t*>* outputDataPtrs,
124 const std::vector<uint8_t*>* outputDataPtrs,
128 const std::vector<int8_t*>* outputDataPtrs,
|
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 1394 std::vector<_Float16*> outputDataPtrs(numOutputs); in executeOperation() local
1396 outputDataPtrs[i] = reinterpret_cast<_Float16*>(operands[outs[i]].buffer); in executeOperation()
1400 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1403 std::vector<float*> outputDataPtrs(numOutputs); in executeOperation() local
1405 outputDataPtrs[i] = reinterpret_cast<float*>(operands[outs[i]].buffer); in executeOperation()
1409 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1412 std::vector<int32_t*> outputDataPtrs(numOutputs); in executeOperation() local
1414 outputDataPtrs[i] = reinterpret_cast<int32_t*>(operands[outs[i]].buffer); in executeOperation()
1418 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1421 std::vector<uint8_t*> outputDataPtrs(numOutputs); in executeOperation() local
[all …]
|