Home
last modified time | relevance | path

Searched refs:composite_devices_ (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/eager/
Dkernel_and_device.cc163 options.composite_devices = composite_devices_; in InstantiateFunc()
435 (composite_devices_.find(input_devices_[i]->name()) == in InputDevice()
436 composite_devices_.end())) { in InputDevice()
Dkernel_and_device.h278 composite_devices_(std::move(composite_devices)), in KernelAndDeviceFunc()
342 absl::flat_hash_map<string, const std::vector<string>*> composite_devices_; variable
Dcontext.cc898 for (const auto& d : composite_devices_) { in FindCompositeDeviceFromName()
929 auto iter = composite_devices_.find(hash_key); in FindOrCreateCompositeDevice()
930 if (iter != composite_devices_.end()) { in FindOrCreateCompositeDevice()
941 composite_devices_.size(), in FindOrCreateCompositeDevice()
949 composite_devices_.emplace(hash_key, std::move(device)); in FindOrCreateCompositeDevice()
Dcontext.h601 composite_devices_ ABSL_GUARDED_BY(composite_devices_mu_);
/external/tensorflow/tensorflow/core/common_runtime/
Dprocess_function_library_runtime.h224 composite_devices_.push_back(d); in AddCompositeDevice()
460 std::vector<CompositeDevice*> composite_devices_ TF_GUARDED_BY(mu_);
Dprocess_function_library_runtime.cc1749 for (auto* d : composite_devices_) (*out_pflr)->AddCompositeDevice(d); in Clone()