Lines Matching refs:output_tmp
822 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
828 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
829 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
830 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
832 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
833 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
840 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
846 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
852 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
858 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
867 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
869 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
886 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
892 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
893 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
894 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
897 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
898 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
905 blockSize, reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
911 blockSize, reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
917 blockSize, reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
923 blockSize, reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
932 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
934 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1107 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1113 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1114 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1115 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1120 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1121 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1129 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1136 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1143 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1150 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1159 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1161 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1179 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1185 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1186 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1187 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()
1203 reinterpret_cast<float*>(output_tmp.buffer), outShape); in executeOperation()
1211 reinterpret_cast<_Float16*>(output_tmp.buffer), outShape); in executeOperation()
1219 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1227 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1236 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1238 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()
1498 RunTimeOperandInfo input_tmp, output_tmp; in executeOperation() local
1504 output_tmp.lifetime = Operand::LifeTime::TEMPORARY_VARIABLE; in executeOperation()
1505 output_tmp.buffer = data_layout ? nullptr : output.buffer; in executeOperation()
1506 output_tmp.length = data_layout ? 0 : output.length; in executeOperation()
1524 !setInfoAndAllocateIfNeeded(&output_tmp, outShape, &result)) { in executeOperation()
1525 if (!data_layout) output.dimensions = output_tmp.dimensions; in executeOperation()
1536 numGroups, activation, reinterpret_cast<float*>(output_tmp.buffer), in executeOperation()
1544 numGroups, activation, reinterpret_cast<_Float16*>(output_tmp.buffer), in executeOperation()
1556 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1564 reinterpret_cast<uint8_t*>(output_tmp.buffer), outShape); in executeOperation()
1576 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1584 reinterpret_cast<int8_t*>(output_tmp.buffer), outShape); in executeOperation()
1589 output_tmp_guard.reset(output_tmp.buffer); in executeOperation()
1591 if (!success || !convertFromNhwc(output, output_tmp, data_layout, &result)) { in executeOperation()