Searched refs:pre_output_tensors (Results 1 – 1 of 1) sorted by relevance
507 std::vector<tensor::TensorPtr> pre_output_tensors; in RunOpAssignOutputNodeMemory() local508 TensorValueToTensor(pre_output_value, &pre_output_tensors); in RunOpAssignOutputNodeMemory()510 if (pre_output_tensors.size() != output_nodes.size()) { in RunOpAssignOutputNodeMemory()511 MS_LOG(EXCEPTION) << "The size of pre output tensors [" << pre_output_tensors.size() in RunOpAssignOutputNodeMemory()530 MS_EXCEPTION_IF_NULL(pre_output_tensors[i]); in RunOpAssignOutputNodeMemory()531 if (pre_output_tensors[i]->device_address() == nullptr) { in RunOpAssignOutputNodeMemory()540 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(pre_output_tensors[i]->dev… in RunOpAssignOutputNodeMemory()543 …AnfAlgo::SetOutputAddr(std::dynamic_pointer_cast<device::DeviceAddress>(pre_output_tensors[i]->dev… in RunOpAssignOutputNodeMemory()