Searched refs:shared_mem_config (Results 1 – 3 of 3) sorted by relevance
381 hipSharedMemConfig shared_mem_config; in ContextGetSharedMemConfig() local384 tensorflow::wrap::hipDeviceGetSharedMemConfig(&shared_mem_config), in ContextGetSharedMemConfig()386 return shared_mem_config; in ContextGetSharedMemConfig()390 GpuContext* context, hipSharedMemConfig shared_mem_config) { in ContextSetSharedMemConfig() argument393 tensorflow::wrap::hipDeviceSetSharedMemConfig(shared_mem_config), in ContextSetSharedMemConfig()
503 CUsharedconfig shared_mem_config; in ContextGetSharedMemConfig() local505 RETURN_IF_CUDA_RES_ERROR(cuCtxGetSharedMemConfig(&shared_mem_config), in ContextGetSharedMemConfig()507 return shared_mem_config; in ContextGetSharedMemConfig()511 GpuContext* context, CUsharedconfig shared_mem_config) { in ContextSetSharedMemConfig() argument513 RETURN_IF_CUDA_RES_ERROR(cuCtxSetSharedMemConfig(shared_mem_config), in ContextSetSharedMemConfig()
254 GpuContext* context, GpuSharedMemConfig shared_mem_config);