Home
last modified time | relevance | path

Searched refs:set_shared_memory_bytes (Results 1 – 4 of 4) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/
H A Dkernel.cc55 void KernelMetadata::set_shared_memory_bytes(int shared_memory_bytes) { in set_shared_memory_bytes() function in stream_executor::KernelMetadata
H A Dkernel.h119 void set_shared_memory_bytes(int shared_memory_bytes);
/aosp_15_r20/external/tensorflow/tensorflow/stream_executor/rocm/
H A Drocm_gpu_executor.cc294 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
H A Dcuda_gpu_executor.cc437 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()