Searched refs:ensure_eigen_gpu_device (Results 1 – 5 of 5) sorted by relevance
55 params.ensure_eigen_gpu_device(); in TEST()
93 sub_params_.ensure_eigen_gpu_device();
453 params.ensure_eigen_gpu_device(); in CallTfKernel()
592 inline void ensure_eigen_gpu_device() { in ensure_eigen_gpu_device() function
348 params_->ensure_eigen_gpu_device(); in OpKernelContext()