• Home
  • Raw
  • Download

Lines Matching refs:input_tmp

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()
896 if (!spaceToDepthPrepare(input_tmp.shape(), blockSize, &outShape) || in executeOperation()
901 switch (input_tmp.type) { in executeOperation()
904 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
910 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
916 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
922 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1107 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1109 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1117 if (!batchToSpacePrepare(input_tmp.shape(), in executeOperation()
1124 switch (input_tmp.type) { in executeOperation()
1127 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1134 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1141 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1148 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1179 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1181 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1190 input_tmp.shape(), reinterpret_cast<const int32_t*>(blockSize.buffer), in executeOperation()
1197 switch (input_tmp.type) { in executeOperation()
1200 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1208 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1216 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1224 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1498 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1500 if (!convertToNhwc(input_tmp, input, input_tmp_guard, data_layout)) { in executeOperation()
1509 Shape inputShape = input_tmp.shape(); in executeOperation()
1521 if (!groupedConvPrepare(input_tmp.shape(), filter.shape(), bias.shape(), padding_left, in executeOperation()
1530 if (input_tmp.type == OperandType::TENSOR_FLOAT32) { in executeOperation()
1532 reinterpret_cast<const float*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1538 } else if (input_tmp.type == OperandType::TENSOR_FLOAT16) { in executeOperation()
1540 reinterpret_cast<const _Float16*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1546 } else if (input_tmp.type == OperandType::TENSOR_QUANT8_ASYMM) { in executeOperation()
1549 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1559 reinterpret_cast<const uint8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1566 } else if (input_tmp.type == OperandType::TENSOR_QUANT8_ASYMM_SIGNED) { in executeOperation()
1569 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()
1579 reinterpret_cast<const int8_t*>(input_tmp.buffer), input_tmp.shape(), in executeOperation()