Lines Matching refs:output_tmp
823 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
829 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
830 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
831 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
833 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
834 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
841 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
847 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
853 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
859 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
868 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
870 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
887 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
893 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
894 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
895 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
898 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
899 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
906 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
912 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
918 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
924 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
933 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
935 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1108 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1114 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1115 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1116 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1121 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1122 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1130 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1137 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1144 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1151 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1160 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1162 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1180 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1186 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1187 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1188 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1194 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1195 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1204 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1212 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1220 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1228 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1237 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1239 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1499 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1505 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1506 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1507 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1525 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1526 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1537 numGroups, activation, reinterpret_cast<float*>(output_tmp.buffer), in executeOperation()
1545 numGroups, activation, reinterpret_cast<_Float16*>(output_tmp.buffer), in executeOperation()
1557 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1565 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1577 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1585 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1590 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1592 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()