Lines Matching refs:output_tmp
798 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
804 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
805 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
806 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
827 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
828 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
839 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
847 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
858 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
867 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
871 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
873 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
943 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
949 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
950 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
951 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
953 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
954 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
961 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
967 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
973 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
982 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
984 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1001 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1007 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
1008 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1009 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1012 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1013 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1020 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1026 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1032 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1041 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1043 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1180 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1186 output_tmp.lifetime = OperandLifeTime::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()
1193 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1194 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1202 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1209 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1216 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1225 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1227 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1245 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1251 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
1252 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1253 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1259 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1260 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1269 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1277 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1285 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1294 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1296 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1583 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1589 output_tmp.lifetime = OperandLifeTime::TEMPORARY_VARIABLE; in executeOperation()
1590 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1591 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1609 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1610 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1621 numGroups, activation, reinterpret_cast<float*>(output_tmp.buffer), in executeOperation()
1629 numGroups, activation, reinterpret_cast<_Float16*>(output_tmp.buffer), in executeOperation()
1640 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1648 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1653 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1655 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()