Searched refs:cuda_config (Results 1 – 3 of 3) sorted by relevance
844 port::StatusOr<CUsharedconfig> cuda_config = in GetDeviceSharedMemoryConfig() local846 if (!cuda_config.ok()) { in GetDeviceSharedMemoryConfig()851 switch (cuda_config.ValueOrDie()) { in GetDeviceSharedMemoryConfig()860 << cuda_config.ValueOrDie(); in GetDeviceSharedMemoryConfig()866 CUsharedconfig cuda_config; in SetDeviceSharedMemoryConfig() local869 cuda_config = CU_SHARED_MEM_CONFIG_DEFAULT_BANK_SIZE; in SetDeviceSharedMemoryConfig()872 cuda_config = CU_SHARED_MEM_CONFIG_FOUR_BYTE_BANK_SIZE; in SetDeviceSharedMemoryConfig()875 cuda_config = CU_SHARED_MEM_CONFIG_EIGHT_BYTE_BANK_SIZE; in SetDeviceSharedMemoryConfig()881 return GpuDriver::ContextSetSharedMemConfig(context_, cuda_config); in SetDeviceSharedMemoryConfig()
519 # create cuda_config.h520 FILE(WRITE ${tensorflow_source_dir}/third_party/gpus/cuda/cuda_config.h
85 tensorflow/third_party/gpus/cuda/cuda_config.h.tpl