Searched defs:device_spec (Results 1 – 9 of 9) sorted by relevance
417 tensorflow::Safe_PyObjectPtr device_spec; member
853 def device_spec(self): member in Context857 def _set_device(self, device_name, device_spec): argument
4272 tensorflow::Safe_PyObjectPtr device_spec; member4298 PyObject* device_spec) {
203 FlatBufferBuilder* builder, const proto::EdgeTpuDeviceSpec& device_spec) { in ConvertEdgeTpuDeviceSpec()
100 NodeBuilder& NodeBuilder::Device(StringPiece device_spec) { in Device()
171 DeviceNameUtils::ParsedName device_spec = device->parsed_name(); in GetTPUDevices() local
126 auto lookup = [&devices](Device device_spec) { in GetTPUDevices()149 for (const auto& device_spec : llvm::make_range( in GetTPUDevices() local
209 NodeDefBuilder& NodeDefBuilder::Device(StringPiece device_spec) { in Device()
380 py::handle device_spec) in EagerContextThreadLocalDataWrapper()