Searched refs:input_tmp (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 876 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
878 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
885 if (!depthToSpacePrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
890 switch (input_tmp.type) { in executeOperation()
893 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
899 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
905 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
911 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
940 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
942 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
[all …]
|