Home
last modified time | relevance | path

Searched defs:virtual_devices (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dreplicate_invariant_op_hoisting.cc117 bool UsesVirtualDevice(const Optional<DictionaryAttr>& virtual_devices, in UsesVirtualDevice()
/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dconvert_graph_test.cc91 auto virtual_devices = in TEST() local
/external/tensorflow/tensorflow/core/protobuf/
Dconfig.proto150 repeated VirtualDevices virtual_devices = 1; field
/external/tensorflow/tensorflow/core/common_runtime/gpu/
Dgpu_device_test.cc83 auto virtual_devices = local
Dgpu_device.cc831 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings() local
1323 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in CreateDevices() local
/external/tensorflow/tensorflow/python/eager/
Dcontext.py1493 def set_logical_device_configuration(self, dev, virtual_devices): argument