Searched refs:input_tmp (Results 1 – 1 of 1) sorted by relevance
| /packages/modules/NeuralNetworks/common/ |
| D | CpuExecutor.cpp | 822 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
824 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
831 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
836 switch (input_tmp.type) { in executeOperation()
839 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
845 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
851 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
857 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
886 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
888 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
[all …]
|