Searched refs:input_tmp (Results 1 – 1 of 1) sorted by relevance
| /frameworks/ml/nn/common/ |
| D | CpuExecutor.cpp | 798 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
800 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
809 Shape inputShape = input_tmp.shape(); in executeOperation()
823 if (!depthwiseConvPrepare(input_tmp.shape(), filter.shape(), bias.shape(), padding_left, in executeOperation()
832 if (input_tmp.type == OperandType::TENSOR_FLOAT32) { in executeOperation()
834 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
840 } else if (input_tmp.type == OperandType::TENSOR_FLOAT16) { in executeOperation()
842 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
848 } else if (input_tmp.type == OperandType::TENSOR_QUANT8_ASYMM) { in executeOperation()
851 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
[all …]
|