Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dreplicate_invariant_op_hoisting.cc87 bool UsesVirtualDevice(const Optional<DictionaryAttr>& virtual_devices, in UsesVirtualDevice()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
H A Dconvert_graph_test.cc81 auto virtual_devices = in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/pluggable_device/
H A Dpluggable_device_factory.cc188 const auto& virtual_devices = device_options.experimental().virtual_devices(); in CreateDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/
H A Dcollective_test_util.cc122 auto* virtual_devices = options.config.mutable_gpu_options() in CreateGPUDevices() local
/aosp_15_r20/external/tensorflow/tensorflow/core/protobuf/
H A Dconfig.proto172 repeated VirtualDevices virtual_devices = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/gpu/
H A Dgpu_device.cc936 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in VerifyVirtualDeviceSettings() local
1443 const auto& virtual_devices = gpu_options.experimental().virtual_devices(); in CreateDevices() local
H A Dgpu_device_test.cc88 auto virtual_devices = local
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dc_api_test_util.cc474 auto virtual_devices = in GetMultiClientServerDef() local
/aosp_15_r20/external/tensorflow/tensorflow/python/eager/
H A Dcontext.py1646 def set_logical_device_configuration(self, dev, virtual_devices): argument
/aosp_15_r20/out/soong/.intermediates/external/tensorflow/tensorflow_core_proto_cpp_lite/android_arm64_armv8-2a_cortex-a55_static_apex33/gen/proto/tensorflow/core/protobuf/
Dconfig.pb.h4386 …ow::GPUOptions_Experimental_VirtualDevices& GPUOptions_Experimental::virtual_devices(int index) co… in virtual_devices() function
4399 GPUOptions_Experimental::virtual_devices() const { in virtual_devices() function