Home
last modified time | relevance | path

Searched defs:CPUDevice (Results 1 – 25 of 260) sorted by relevance

1234567891011

/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_transpose_op.cc128 typedef Eigen::ThreadPoolDevice CPUDevice; in DoTranspose() typedef
154 typedef Eigen::ThreadPoolDevice CPUDevice; in DoTranspose() typedef
H A Dmkl_qmatmul_op.cc555 #define TEMPLATE_ARGS(CPUDevice, quint8, qint8, bias_type, output_type, \ argument
576 #define TEMPLATE_ARGS(CPUDevice, quint8, qint8, bias_type, output_type, \ argument
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/
H A Dimage_ops.cc34 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
45 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dtranspose_op.cc191 typedef Eigen::ThreadPoolDevice CPUDevice; in DoTranspose() typedef
200 typedef Eigen::ThreadPoolDevice CPUDevice; in DoTranspose() typedef
H A Dinplace_ops.cc28 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
362 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dconv_3d.h40 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Daggregate_ops_cpu.h24 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Drandom_op.h31 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_tstring.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_variant.cc24 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_uint64.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_int64.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_complex64.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_int32.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_bool.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_uint32.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_complex128.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_int8.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_uint8.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_bfloat16.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_float.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
H A Dtile_functor_cpu_int16.cc23 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/linalg/
H A Dmatrix_band_part_op.cc43 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
134 typedef Eigen::ThreadPoolDevice CPUDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/examples/custom_ops_doc/multiplex_2/
H A Dmultiplex_2_kernel.cc22 using CPUDevice = Eigen::ThreadPoolDevice; typedef
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dregister_types_traits.h21 typedef Eigen::ThreadPoolDevice CPUDevice; typedef

1234567891011