Searched refs:shared_mem_config (Results 1 – 3 of 3) sorted by relevance
403 hipSharedMemConfig shared_mem_config; in ContextGetSharedMemConfig() local406 tensorflow::wrap::hipDeviceGetSharedMemConfig(&shared_mem_config); in ContextGetSharedMemConfig()415 return shared_mem_config; in ContextGetSharedMemConfig()419 GpuContext* context, hipSharedMemConfig shared_mem_config) { in ContextSetSharedMemConfig() argument422 tensorflow::wrap::hipDeviceSetSharedMemConfig(shared_mem_config); in ContextSetSharedMemConfig()426 << ", config: " << shared_mem_config in ContextSetSharedMemConfig()
515 CUsharedconfig shared_mem_config; in ContextGetSharedMemConfig() local517 CUresult result = cuCtxGetSharedMemConfig(&shared_mem_config); in ContextGetSharedMemConfig()528 return shared_mem_config; in ContextGetSharedMemConfig()532 GpuContext* context, CUsharedconfig shared_mem_config) { in ContextSetSharedMemConfig() argument534 CUresult result = cuCtxSetSharedMemConfig(shared_mem_config); in ContextSetSharedMemConfig()540 << ", config: " << shared_mem_config in ContextSetSharedMemConfig()
197 GpuContext* context, GpuSharedMemConfig shared_mem_config);