Home
last modified time | relevance | path

Searched refs:SharedMemoryConfig (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_gpu_executor.cc724 SharedMemoryConfig GpuExecutor::GetDeviceSharedMemoryConfig() { in GetDeviceSharedMemoryConfig()
729 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
734 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
736 return SharedMemoryConfig::kFourByte; in GetDeviceSharedMemoryConfig()
738 return SharedMemoryConfig::kEightByte; in GetDeviceSharedMemoryConfig()
746 SharedMemoryConfig config) { in SetDeviceSharedMemoryConfig()
749 case SharedMemoryConfig::kDefault: in SetDeviceSharedMemoryConfig()
752 case SharedMemoryConfig::kFourByte: in SetDeviceSharedMemoryConfig()
755 case SharedMemoryConfig::kEightByte: in SetDeviceSharedMemoryConfig()
/external/tensorflow/tensorflow/compiler/xla/service/interpreter/
Dexecutor.h178 SharedMemoryConfig GetDeviceSharedMemoryConfig() override { in GetDeviceSharedMemoryConfig()
179 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
182 port::Status SetDeviceSharedMemoryConfig(SharedMemoryConfig config) override { in SetDeviceSharedMemoryConfig()
/external/tensorflow/tensorflow/stream_executor/host/
Dhost_gpu_executor.h160 SharedMemoryConfig GetDeviceSharedMemoryConfig() override { in GetDeviceSharedMemoryConfig()
163 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
166 port::Status SetDeviceSharedMemoryConfig(SharedMemoryConfig config) override { in SetDeviceSharedMemoryConfig()
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_gpu_executor.cc843 SharedMemoryConfig GpuExecutor::GetDeviceSharedMemoryConfig() { in GetDeviceSharedMemoryConfig()
848 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
853 return SharedMemoryConfig::kDefault; in GetDeviceSharedMemoryConfig()
855 return SharedMemoryConfig::kFourByte; in GetDeviceSharedMemoryConfig()
857 return SharedMemoryConfig::kEightByte; in GetDeviceSharedMemoryConfig()
865 SharedMemoryConfig config) { in SetDeviceSharedMemoryConfig()
868 case SharedMemoryConfig::kDefault: in SetDeviceSharedMemoryConfig()
871 case SharedMemoryConfig::kFourByte: in SetDeviceSharedMemoryConfig()
874 case SharedMemoryConfig::kEightByte: in SetDeviceSharedMemoryConfig()
/external/tensorflow/tensorflow/stream_executor/
Dshared_memory_config.h26 enum class SharedMemoryConfig { enum
Dstream_executor_pimpl.cc276 SharedMemoryConfig StreamExecutor::GetDeviceSharedMemoryConfig() { in GetDeviceSharedMemoryConfig()
281 SharedMemoryConfig config) { in SetDeviceSharedMemoryConfig()
282 if (config != SharedMemoryConfig::kDefault && in SetDeviceSharedMemoryConfig()
283 config != SharedMemoryConfig::kFourByte && in SetDeviceSharedMemoryConfig()
284 config != SharedMemoryConfig::kEightByte) { in SetDeviceSharedMemoryConfig()
Dstream_executor_internal.h265 virtual SharedMemoryConfig GetDeviceSharedMemoryConfig() = 0;
267 SharedMemoryConfig config) = 0;
Dstream_executor_pimpl.h339 SharedMemoryConfig GetDeviceSharedMemoryConfig();
343 port::Status SetDeviceSharedMemoryConfig(SharedMemoryConfig config);
/external/tensorflow/tensorflow/stream_executor/gpu/
Dgpu_executor.h190 SharedMemoryConfig GetDeviceSharedMemoryConfig() override;
192 port::Status SetDeviceSharedMemoryConfig(SharedMemoryConfig config) override;