Searched defs:device_spec (Results 1 – 14 of 14) sorted by relevance
415 tensorflow::Safe_PyObjectPtr device_spec; member
978 def device_spec(self): member in Context982 def _set_device(self, device_name, device_spec): argument
4100 tensorflow::Safe_PyObjectPtr device_spec; member4126 PyObject* device_spec) {
38 for (const auto& device_spec : device_specs) { in AttachCostPerDevice() local
64 struct io_mmc_dev_spec *device_spec = in mmc_dev_open() local
105 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device()
171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices() local
228 DeviceNameUtils::ParsedName device_spec; in Run() local
122 auto lookup = [&devices](Device device_spec) { in GetTPUDevices()145 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
271 FlatBufferBuilder* builder, const proto::EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
281 const EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device()
271 for (auto& device_spec : specified_device_specs) { in ProcessTargetDevices() local
489 py::handle device_spec) in EagerContextThreadLocalDataWrapper()