Home
last modified time | relevance | path

Searched refs:set_shared_memory_per_core (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Ddevice_description.h413 void set_shared_memory_per_core(int64_t value) { in set_shared_memory_per_core() function
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc959 builder.set_shared_memory_per_core( in CreateDeviceDescription()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc1107 builder.set_shared_memory_per_core( in CreateDeviceDescription()