| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/ |
| H A D | gpu_device_factory.cc | 29 class GPUDevice : public BaseGPUDevice { class
31 GPUDevice(const SessionOptions& options, const string& name, in GPUDevice() function in tensorflow::GPUDevice
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/ |
| H A D | cholesky_op_gpu.cu.cc | 65 typedef Eigen::GpuDevice GPUDevice; typedef
86 typedef Eigen::GpuDevice GPUDevice; typedef
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | transpose_op.cc | 223 typedef Eigen::GpuDevice GPUDevice; in TF_CALL_ALL_TYPES() typedef
231 typedef Eigen::GpuDevice GPUDevice; in DoTranspose() typedef
|
| H A D | inplace_ops.cc | 169 typedef Eigen::GpuDevice GPUDevice; typedef
445 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | scatter_functor_gpu.cu.h | 30 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | snapshot_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | avgpooling_op.h | 46 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | strided_slice_op_gpu_impl.h | 31 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | broadcast_to_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | cross_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | cwise_ops_gpu_gradients.cu.h | 36 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | argmax_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | slice_op_gpu.cu.cc | 28 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | cwise_ops_gpu_common.cu.h | 35 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | list_kernels.cu.cc | 37 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | conv_2d_gpu.h | 39 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | cast_op_impl_double.cc | 21 typedef Eigen::GpuDevice GPUDevice; typedef
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | adjust_hue_op.h | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | adjust_saturation_op.h | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | colorspace_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | extract_image_patches_op_gpu.cu.cc | 26 typedef Eigen::GpuDevice GPUDevice; typedef
|
| H A D | extract_volume_patches_op_gpu.cu.cc | 25 typedef Eigen::GpuDevice GPUDevice; typedef
|
| /aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/ |
| H A D | multiplex_2_kernel.cu.cc | 24 using GPUDevice = Eigen::GpuDevice; typedef
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | gru_ops_gpu.cu.cc | 24 typedef Eigen::GpuDevice GPUDevice; typedef
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/framework/ |
| H A D | register_types_traits.h | 22 typedef Eigen::GpuDevice GPUDevice; typedef
|