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.h725 bool is_cpu = std::is_same<Device, CPUDevice>::value;
726 OP_REQUIRES(ctx, is_cpu,
/external/tensorflow/tensorflow/core/common_runtime/eager/
Dtensor_handle.cc1056 bool is_cpu = device_ != nullptr; in DebugString() local
1061 is_cpu ? absl::visit([](auto& data) { return data.DebugString(); }, data_) in DebugString()
/external/ImageMagick/MagickCore/
Dopencl.c1052 static double RunOpenCLBenchmark(MagickBooleanType is_cpu) in RunOpenCLBenchmark() argument
1097 if (is_cpu == MagickFalse) in RunOpenCLBenchmark()