Searched defs:new_device_tensor (Results 1 – 6 of 6) sorted by relevance
220 DeviceTensor *new_device_tensor = nullptr; in MergeDynamiclenDeviceAddress() local352 …DeviceTensorPtr new_device_tensor = device_context->device_res_manager_->CreateDeviceAddress(new_k… in CopyDeviceAddress() local
180 DeviceTensor *new_device_tensor = nullptr; in GatherInput() local
594 …const auto &new_device_tensor = device_context->device_res_manager_->CreateDeviceAddress(kernel_te… in MergeDeviceAddress() local682 …const auto &new_device_tensor = device_context->device_res_manager_->CreateDeviceAddress(kernel_te… in MergeEmptyAddressDeviceAddress() local
718 auto &new_device_tensor = copy_input_device_tensors_[input_data_index]; in CopyInputDeviceTensor() local1268 for (auto &new_device_tensor : need_refreshed_device_tensors) { in RefreshDeviceTensorCopyStore() local1289 for (auto &new_device_tensor : need_refreshed_device_tensors) { in RefreshDeviceTensorCopyStore() local
296 …auto new_device_tensor = real_device_context->device_res_manager_->CreateDeviceAddress(kernel_tens… in HandleBackoffValueNode() local
3151 … auto new_device_tensor = device_context->device_res_manager_->CreateDeviceAddress(kernel_tensor); in PersistDeviceTensorForRootGraphControlNode() local