Searched defs:input_device_tensor (Results 1 – 3 of 3) sorted by relevance
33 DeviceTensor *input_device_tensor, const KernelTransformType &type) { in UpdateShape()48 inline bool InputDataNoNeedCopy(const AnfNodePtr &input_node, DeviceTensor *input_device_tensor, in InputDataNoNeedCopy()294 auto input_device_tensor = DeviceTensorStore::GetInstance() in FetchPersistentDeviceTensor() local626 DeviceTensor *input_device_tensor, in CopyInputDataPersistedHandle()706 auto &input_device_tensor = input_device_tensors_[i]; in CopyInputData() local
152 const auto &input_device_tensor = AnfAlgo::GetPrevNodeMutableOutputAddr(kernel_, i, false); in InitInputInfo() local1265 auto &input_device_tensor = input_device_tensors_[ref_input_index]; in RefreshDeviceTensorCopyStore() local1314 void KernelActor::SetInputDeviceTensor(DeviceTensor *input_device_tensor, size_t input_index) { in SetInputDeviceTensor()
257 bool ExitActor::IsNeedCopyDeviceAddress(DeviceTensor *const input_device_tensor, size_t index) { in IsNeedCopyDeviceAddress()322 auto &input_device_tensor = input_device_tensors_[i]; in CopyDeviceAddress() local