Searched defs:host_tensor_address (Results 1 – 1 of 1) sorted by relevance
302 …auto host_tensor_address = std::dynamic_pointer_cast<DeviceTensor>(input_tensor->device_address()); in PrepareDataForStepMode() local414 auto host_tensor_address = std::dynamic_pointer_cast<DeviceTensor>(tensor->device_address()); in PrepareDataForWeightNode() local