Searched refs:outputDataPtrs (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 29 const std::vector<Scalar*>* outputDataPtrs, in splitGeneric() argument
44 for (int i = 0; i < outputDataPtrs->size(); ++i) { in splitGeneric()
46 memcpy(outputDataPtrs->at(i) + k * copySize, inputPtr, copySize * sizeof(Scalar)); in splitGeneric()
55 const std::vector<_Float16*>* outputDataPtrs, in splitFloat16() argument
58 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
62 const std::vector<float*>* outputDataPtrs, in splitFloat32() argument
65 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
69 const std::vector<uint8_t*>* outputDataPtrs, in splitQuant8() argument
72 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
76 const std::vector<int32_t*>* outputDataPtrs, in splitInt32() argument
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 126 const std::vector<_Float16*>* outputDataPtrs,
130 const std::vector<float*>* outputDataPtrs,
134 const std::vector<int32_t*>* outputDataPtrs,
138 const std::vector<uint8_t*>* outputDataPtrs,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1487 std::vector<_Float16*> outputDataPtrs(numOutputs); in executeOperation() local
1489 outputDataPtrs[i] = reinterpret_cast<_Float16*>(mOperands[outs[i]].buffer); in executeOperation()
1493 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1496 std::vector<float*> outputDataPtrs(numOutputs); in executeOperation() local
1498 outputDataPtrs[i] = reinterpret_cast<float*>(mOperands[outs[i]].buffer); in executeOperation()
1502 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1505 std::vector<int32_t*> outputDataPtrs(numOutputs); in executeOperation() local
1507 outputDataPtrs[i] = reinterpret_cast<int32_t*>(mOperands[outs[i]].buffer); in executeOperation()
1511 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1514 std::vector<uint8_t*> outputDataPtrs(numOutputs); in executeOperation() local
[all …]
|