Searched refs:kernel_metadata (Results 1 – 3 of 3) sorted by relevance
278 KernelMetadata kernel_metadata; in GetKernel() local279 TF_RETURN_IF_ERROR(GetKernelMetadata(rocm_kernel, &kernel_metadata)); in GetKernel()280 kernel->set_metadata(kernel_metadata); in GetKernel()286 KernelMetadata* kernel_metadata) { in GetKernelMetadata() argument289 kernel_metadata->set_registers_per_thread(value); in GetKernelMetadata()292 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()
308 KernelMetadata kernel_metadata; in GetKernel() local309 TF_RETURN_IF_ERROR(GetKernelMetadata(cuda_kernel, &kernel_metadata)); in GetKernel()310 kernel->set_metadata(kernel_metadata); in GetKernel()387 KernelMetadata* kernel_metadata) { in GetKernelMetadata() argument391 kernel_metadata->set_registers_per_thread(value); in GetKernelMetadata()396 kernel_metadata->set_shared_memory_bytes(value); in GetKernelMetadata()
263 KernelMetadata* kernel_metadata);