Searched defs:host_tensor_address (Results 1 – 1 of 1) sorted by relevance
1181 … const device::DeviceAddressPtr &host_tensor_address, in CopyDataFromDeviceTensorStore()1256 auto host_tensor_address = std::dynamic_pointer_cast<DeviceTensor>(tensor->device_address()); in PrepareDataForWeightNode() local1371 auto host_tensor_address = std::dynamic_pointer_cast<DeviceTensor>(tensor->device_address()); in PrepareDeviceTensorStoreForControlNode() local