Lines Matching refs:input_tmp
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()
950 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
955 switch (input_tmp.type) { in executeOperation()
958 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
964 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
970 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
976 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1164 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1166 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1174 if (!batchToSpacePrepare(input_tmp.shape(), in executeOperation()
1181 switch (input_tmp.type) { in executeOperation()
1184 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1191 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1198 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1205 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1236 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1238 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1247 input_tmp.shape(), reinterpret_cast<const int32_t*>(blockSize.buffer), in executeOperation()
1254 switch (input_tmp.type) { in executeOperation()
1257 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1265 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1273 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1281 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1555 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1557 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1566 Shape inputShape = input_tmp.shape(); in executeOperation()
1578 if (!groupedConvPrepare(input_tmp.shape(), filter.shape(), bias.shape(), padding_left, in executeOperation()
1587 if (input_tmp.type == OperandType::TENSOR_FLOAT32) { in executeOperation()
1589 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1595 } else if (input_tmp.type == OperandType::TENSOR_FLOAT16) { in executeOperation()
1597 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1603 } else if (input_tmp.type == OperandType::TENSOR_QUANT8_ASYMM) { in executeOperation()
1606 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1615 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1622 } else if (input_tmp.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) { in executeOperation()
1625 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1634 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()