Home
last modified time | relevance | path

Searched refs:ensure_eigen_gpu_device (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/tfrt/fallback/
Dop_kernel_runner_test.cc55 params.ensure_eigen_gpu_device(); in TEST()
/external/tensorflow/tensorflow/core/common_runtime/
Dcollective_util.cc93 sub_params_.ensure_eigen_gpu_device();
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dlight_outside_compilation.cc453 params.ensure_eigen_gpu_device(); in CallTfKernel()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h592 inline void ensure_eigen_gpu_device() { in ensure_eigen_gpu_device() function
Dop_kernel.cc348 params_->ensure_eigen_gpu_device(); in OpKernelContext()