Searched refs:workspace_data (Results 1 – 2 of 2) sorted by relevance
74 se::DeviceMemoryBase workspace_data = in ExecuteOnStream() local85 info_data, se::DeviceMemory<float>(workspace_data))); in ExecuteOnStream()91 se::DeviceMemory<double>(workspace_data))); in ExecuteOnStream()97 info_data, se::DeviceMemory<std::complex<float>>(workspace_data))); in ExecuteOnStream()103 info_data, se::DeviceMemory<std::complex<double>>(workspace_data))); in ExecuteOnStream()
80 std::shared_ptr<stream> fwd_stream, U* workspace_data) { in Execute() argument99 if (workspace_data != nullptr) { in Execute()100 context_.ws_mem->set_data_handle(workspace_data, *fwd_stream); in Execute()116 if (workspace_data != nullptr) { in Execute()117 context_.ws_mem->set_data_handle(workspace_data); in Execute()136 if (workspace_data != nullptr) { in Execute()