Searched refs:output_tmp (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 878 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
884 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
885 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
886 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
888 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
889 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
896 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
902 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
908 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
914 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
[all …]
|