Searched refs:matched_device (Results 1 – 1 of 1) sorted by relevance
544 tensorflow::Device* matched_device = nullptr; in PYBIND11_MODULE() local555 if (matched_device != nullptr) { in PYBIND11_MODULE()561 device_name, matched_device->name(), device->name()) in PYBIND11_MODULE()564 matched_device = device; in PYBIND11_MODULE()568 if (matched_device == nullptr) { in PYBIND11_MODULE()575 tensorflow::Allocator* allocator = matched_device->GetAllocator(attrs); in PYBIND11_MODULE()586 matched_device->name()) in PYBIND11_MODULE()