Searched refs:default_device_ (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 61 TF_CHECK_OK(dev_mgr_->LookupDevice(*kDefaultDeviceName, &default_device_));
67 Device* default_device_; member in tensorflow::__anon5d7062b30111::BufRendezvousTest
89 *kDefaultKey, default_device_, fake_device_context_, &a_, aa_, in TEST_F()
104 EXPECT_EQ(h->prod_dev, default_device_); in TEST_F()
130 EXPECT_EQ(h->prod_dev, default_device_); in TEST_F()
138 *kDefaultKey, default_device_, fake_device_context_, &a_, aa_, in TEST_F()
155 *kDefaultKey, default_device_, fake_device_context_, &a_, aa_, in TEST_F()
161 *kDefaultKey, default_device_, fake_device_context_, &a_, aa_, in TEST_F()
205 "key1", default_device_, fake_device_context_, &a_, aa_, in TEST_F()
238 "key1", default_device_, fake_device_context_, &a_, aa_, in TEST_F()
[all …]
|
| D | inspecting_placer.cc | 122 default_device_(default_device), in InspectingPlacer()
150 default_device_, allow_soft_placement_, log_device_placement_); in ComputeIOColocationGroups()
|
| D | inspecting_placer.h | 86 const Device* default_device_; variable
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | tf_device_assignment.cc | 32 default_device_ = std::string(default_device); in SimpleTFDeviceAssignmentPass()
42 op->setAttr("device", builder.getStringAttr(default_device_)); in runOnFunction()
47 op->setAttr("device", builder.getStringAttr(default_device_)); in runOnFunction()
53 Option<std::string> default_device_{ member in mlir::TF::__anoned113ffe0111::SimpleTFDeviceAssignmentPass
|
| /external/tensorflow/tensorflow/compiler/xla/python/ |
| D | jax_jit.cc | 912 xla::PjRtDevice* default_device_ = nullptr; member in jax::__anone04173081411::CompiledFunction
1110 if (!default_device_) { in Call()
1120 if (!default_device_) { in Call()
1131 default_device_ = default_pydevice_.contents; in Call()
1132 if (!default_device_) { // UPTC in Call()
1141 CHECK(default_device_); in Call()
1153 default_device_, is_committed_, arguments) in Call()
|