Searched defs:tensor_address (Results 1 – 9 of 9) sorted by relevance
187 auto tensor_address = tensor->device_address(); in LoadInputData() local
67 … auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in UpdateInputTensorFromDevice() local106 …auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(input_tensor->device_addres… in SetDeviceAddress() local
396 const auto &tensor_address = input_tensor->device_address(); in ReplaceGraphParameterProperties() local
314 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in CheckIfNeedSync() local
330 auto tensor_address = tensor->device_address(); in BindInputTensorAddressPtr() local
997 auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in AssignExtraStaticMem() local1588 … auto tensor_address = std::dynamic_pointer_cast<device::DeviceAddress>(tensor->device_address()); in InitGraphInputTensors() local
940 auto tensor_address = std::static_pointer_cast<device::DeviceAddress>(device_sync); in CreateInputAddressForViewOp() local
460 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in UpdateDeviceAddressForDataNode() local
322 auto tensor_address = std::dynamic_pointer_cast<DeviceTensor>(res_tensor->device_address()); in GetUserDataFromAddress() local