Searched defs:default_device_ (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ | ||
| D | tf_device_assignment.cc | 53 Option<std::string> default_device_{ member in mlir::TF::__anoned113ffe0111::SimpleTFDeviceAssignmentPass |
| /external/tensorflow/tensorflow/core/common_runtime/ | ||
| D | inspecting_placer.h | 86 const Device* default_device_; variable |
| D | buf_rendezvous_test.cc | 67 Device* default_device_; member in tensorflow::__anon5d7062b30111::BufRendezvousTest |
| /external/tensorflow/tensorflow/compiler/xla/python/ | ||
| D | jax_jit.cc | 912 xla::PjRtDevice* default_device_ = nullptr; member in jax::__anone04173081411::CompiledFunction |