Home
last modified time | relevance | path

Searched refs:is_cpu (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/tpu/
Dfeature_column_v2.py929 is_cpu = self._embedding_lookup_device == EmbeddingDevice.CPU
930 is_cpu = is_cpu or _is_running_on_cpu()
931 if is_cpu:
945 is_cpu = self._embedding_lookup_device == EmbeddingDevice.CPU
946 is_cpu = is_cpu or _is_running_on_cpu()
947 if is_cpu:
981 is_cpu = self._embedding_lookup_device == EmbeddingDevice.CPU
982 is_cpu = is_cpu or _is_running_on_cpu()
983 if is_cpu:
1058 is_cpu = self._embedding_lookup_device == EmbeddingDevice.CPU
[all …]
/external/tensorflow/tensorflow/core/kernels/
Dmatmul_op_impl.h744 bool is_cpu = std::is_same<Device, CPUDevice>::value;
745 OP_REQUIRES(ctx, is_cpu,
/external/ImageMagick/MagickCore/
Dopencl.c1052 static double RunOpenCLBenchmark(MagickBooleanType is_cpu) in RunOpenCLBenchmark() argument
1097 if (is_cpu == MagickFalse) in RunOpenCLBenchmark()
/external/eigen/unsupported/Eigen/CXX11/src/Tensor/
DTensorDeviceSycl.h92 (device.is_cpu() && platform_name.find("amd") != std::string::npos &&