929 	return psp_ta_init_shared_buf(psp, &psp->xgmi_context.context.mem_context,  in psp_xgmi_init_shared_buf()
975 				 psp->xgmi_context.context.mem_context.shared_mc_addr,  in psp_xgmi_load()
1037 	psp_ta_free_shared_buf(&psp->xgmi_context.context.mem_context);  in psp_xgmi_terminate()
1068 	xgmi_cmd = (struct ta_xgmi_shared_memory *)(psp->xgmi_context.context.mem_context.shared_buf);  in psp_xgmi_initialize()
1083 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_hive_id()
1103 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_node_id()
1184 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_get_topology_info()
1264 	xgmi_cmd = (struct ta_xgmi_shared_memory *)psp->xgmi_context.context.mem_context.shared_buf;  in psp_xgmi_set_topology_info()
1285 	return psp_ta_init_shared_buf(psp, &psp->ras_context.context.mem_context,  in psp_ras_init_shared_buf()
1303 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_load()
1315 				 psp->ras_context.context.mem_context.shared_mc_addr,  in psp_ras_load()
1366 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_invoke()
1433 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_enable_features()
1470 	psp_ta_free_shared_buf(&psp->ras_context.context.mem_context);  in psp_ras_terminate()
1556 	ras_cmd = (struct ta_ras_shared_memory *)psp->ras_context.context.mem_context.shared_buf;  in psp_ras_trigger_error()
1578 	return psp_ta_init_shared_buf(psp, &psp->hdcp_context.context.mem_context,  in psp_hdcp_init_shared_buf()
1601 				 psp->hdcp_context.context.mem_context.shared_mc_addr,  in psp_hdcp_load()
1689 		if (psp->hdcp_context.context.mem_context.shared_buf)  in psp_hdcp_terminate()
1703 	psp_ta_free_shared_buf(&psp->hdcp_context.context.mem_context);  in psp_hdcp_terminate()
1712 	return psp_ta_init_shared_buf(psp, &psp->dtm_context.context.mem_context,  in psp_dtm_init_shared_buf()
1734 				 psp->dtm_context.context.mem_context.shared_mc_addr,  in psp_dtm_load()
1823 		if (psp->dtm_context.context.mem_context.shared_buf)  in psp_dtm_terminate()
1837 	psp_ta_free_shared_buf(&psp->dtm_context.context.mem_context);  in psp_dtm_terminate()
1846 	return psp_ta_init_shared_buf(psp, &psp->rap_context.context.mem_context,  in psp_rap_init_shared_buf()
1862 				 psp->rap_context.context.mem_context.shared_mc_addr,  in psp_rap_load()
1923 		psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_rap_initialize()
1948 	psp_ta_free_shared_buf(&psp->rap_context.context.mem_context);  in psp_rap_terminate()
1968 		  psp->rap_context.context.mem_context.shared_buf;  in psp_rap_invoke()
1992 		psp, &psp->securedisplay_context.context.mem_context,  in psp_securedisplay_init_shared_buf()
2007 				 psp->securedisplay_context.context.mem_context.shared_mc_addr,  in psp_securedisplay_load()
2071 		psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_securedisplay_initialize()
2108 	psp_ta_free_shared_buf(&psp->securedisplay_context.context.mem_context);  in psp_securedisplay_terminate()