Searched refs:outputDataPtrs (Results 1 – 3 of 3) sorted by relevance
| /frameworks/ml/nn/common/operations/ |
| D | Split.cpp | 31 const std::vector<Scalar*>* outputDataPtrs, in splitGeneric() argument
46 for (int i = 0; i < outputDataPtrs->size(); ++i) { in splitGeneric()
48 memcpy(outputDataPtrs->at(i) + k * copySize, inputPtr, copySize * sizeof(Scalar)); in splitGeneric()
57 const std::vector<_Float16*>* outputDataPtrs, in splitFloat16() argument
60 return splitGeneric<_Float16>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat16()
64 const std::vector<float*>* outputDataPtrs, in splitFloat32() argument
67 return splitGeneric<float>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitFloat32()
71 const std::vector<uint8_t*>* outputDataPtrs, in splitQuant8() argument
74 return splitGeneric<uint8_t>(inputData, inputShape, axis, outputDataPtrs, outputShapes); in splitQuant8()
78 const std::vector<int8_t*>* outputDataPtrs, in splitQuant8Signed() argument
[all …]
|
| /frameworks/ml/nn/common/include/ |
| D | Operations.h | 125 const std::vector<_Float16*>* outputDataPtrs,
129 const std::vector<float*>* outputDataPtrs,
133 const std::vector<int32_t*>* outputDataPtrs,
137 const std::vector<uint8_t*>* outputDataPtrs,
141 const std::vector<int8_t*>* outputDataPtrs,
|
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 1450 std::vector<_Float16*> outputDataPtrs(numOutputs); in executeOperation() local
1452 outputDataPtrs[i] = reinterpret_cast<_Float16*>(operands[outs[i]].buffer); in executeOperation()
1456 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1459 std::vector<float*> outputDataPtrs(numOutputs); in executeOperation() local
1461 outputDataPtrs[i] = reinterpret_cast<float*>(operands[outs[i]].buffer); in executeOperation()
1465 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1468 std::vector<int32_t*> outputDataPtrs(numOutputs); in executeOperation() local
1470 outputDataPtrs[i] = reinterpret_cast<int32_t*>(operands[outs[i]].buffer); in executeOperation()
1474 input.shape(), axis, &outputDataPtrs, outputShapes); in executeOperation()
1477 std::vector<uint8_t*> outputDataPtrs(numOutputs); in executeOperation() local
[all …]
|