Searched defs:device_spec (Results 1 – 13 of 13) sorted by relevance
417 tensorflow::Safe_PyObjectPtr device_spec; member
917 def device_spec(self): member in Context921 def _set_device(self, device_name, device_spec): argument
4365 tensorflow::Safe_PyObjectPtr device_spec; member4391 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
171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices() local
107 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device()
126 auto lookup = [&devices](Device device_spec) { in GetTPUDevices()149 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
241 FlatBufferBuilder* builder, const proto::EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
255 const EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device()
268 for (auto& device_spec : specified_device_specs) { in ProcessTargetDevices() local
450 py::handle device_spec) in EagerContextThreadLocalDataWrapper()