Searched refs:input_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
880 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
887 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
892 switch (input_tmp.type) { in executeOperation()
895 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
901 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
907 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
913 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
942 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
944 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
[all …]
|