Lines Matching full:tensor_out
246 auto tensor_out = args[handles.inputs->count + i]->toTensor(); in execute() local
250 tensor_out, in execute()
254 if (tensor_out.scalar_type() == ScalarType::Char and in execute()
259 tensor_out.mutable_data_ptr<char>(), in execute()
260 tensor_out.size(1), in execute()
261 tensor_out.size(2), in execute()
262 tensor_out.size(3)); in execute()
264 for (int j = 0; j < tensor_out.numel(); j++) { in execute()
265 if (tensor_out.scalar_type() == ScalarType::Char) { in execute()
267 tensor_out.mutable_data_ptr<char>()[j] = output_address[j]; in execute()
270 tensor_out.mutable_data_ptr<int>()[j] = output_address[j]; in execute()