Home
last modified time | relevance | path

Searched refs:device_attributes (Results 1 – 25 of 30) sorted by relevance

12

/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc49 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
50 : Device(nullptr, device_attributes) {} in FakeDevice()
63 DeviceAttributes device_attributes; in Make() local
64 device_attributes.set_name(name); in Make()
65 device_attributes.set_device_type(DeviceType(type).type()); in Make()
66 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in Make()
/external/tensorflow/tensorflow/core/common_runtime/
Dcomposite_device.cc68 DeviceAttributes device_attributes; in MakeDevice() local
69 device_attributes.set_name(device_name); in MakeDevice()
70 device_attributes.set_device_type(kCompositeDeviceType); in MakeDevice()
73 new CompositeDevice(device_attributes, underlying_devices)); in MakeDevice()
Dcomposite_device.h59 CompositeDevice(const DeviceAttributes& device_attributes, in CompositeDevice() argument
61 : Device(/*env=*/nullptr, device_attributes), in CompositeDevice()
Dconstant_folding_test.cc98 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
99 : Device(nullptr, device_attributes) {} in FakeDevice()
107 DeviceAttributes device_attributes; in Make() local
108 device_attributes.set_name(name); in Make()
109 device_attributes.set_device_type(DeviceType(type).type()); in Make()
110 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in Make()
Dcolocation_graph.cc934 bool IsSupportedDeviceType(const DeviceAttributes& device_attributes, in IsSupportedDeviceType() argument
936 if (DeviceType(device_attributes.device_type()) == supported_type) { in IsSupportedDeviceType()
939 return IsCompositeDevice(device_attributes.device_type()); in IsSupportedDeviceType()
Dplacer_test.cc87 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
88 : Device(nullptr, device_attributes) {} in FakeDevice()
97 DeviceAttributes device_attributes; in MakeDevice() local
98 device_attributes.set_name(name); in MakeDevice()
99 device_attributes.set_device_type(device_type); in MakeDevice()
100 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeDevice()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Ddevice_util_test.cc43 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
44 : Device(nullptr, device_attributes) {} in FakeDevice()
53 DeviceAttributes device_attributes; in Make() local
54 device_attributes.set_name(name); in Make()
55 device_attributes.set_device_type(parsed_name.type); in Make()
56 device_attributes.set_physical_device_desc(desc); in Make()
57 return std::make_unique<FakeDevice>(device_attributes); in Make()
/external/tensorflow/tensorflow/core/distributed_runtime/
Dsession_mgr_test.cc28 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
29 : Device(nullptr, device_attributes) {} in FakeDevice()
37 DeviceAttributes device_attributes; in MakeCPU() local
38 device_attributes.set_name(name); in MakeCPU()
39 device_attributes.set_device_type(DeviceType("FakeCPU").type()); in MakeCPU()
40 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeCPU()
Dremote_device.cc63 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes, in AsRemoteDevices() argument
66 for (const auto& da : device_attributes) { in AsRemoteDevices()
113 for (const DeviceAttributes& da : call->resp.device_attributes()) { in NewRemoteDevices()
Dsession_mgr.h54 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
66 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
Dremote_device.h42 const protobuf::RepeatedPtrField<DeviceAttributes>& device_attributes,
Dworker_cache_partial.cc80 for (auto& dev_attr : resp.device_attributes()) { in RefreshDeviceStatus()
Dcollective_param_resolver_distributed.cc147 request->device_attributes(), cp, cancel_mgr, in CompleteGroupAsync()
242 if (resp.device_attributes().empty()) { in UpdateGroupCache()
251 for (const DeviceAttributes& device : resp.device_attributes()) { in UpdateGroupCache()
Dcollective_rma_distributed.cc224 for (const DeviceAttributes& da : resp->device_attributes()) { in CheckPeerHealth()
Dcollective_param_resolver_distributed_test.cc118 for (const auto& it : resp.device_attributes()) { in GetDeviceLocalityAsync()
Dcollective_rma_distributed_test.cc171 for (const auto& it : resp.device_attributes()) { in GetDeviceLocalityAsync()
/external/tensorflow/tensorflow/core/framework/
Ddevice.cc26 Device::Device(Env* env, const DeviceAttributes& device_attributes) in Device() argument
27 : DeviceBase(env), device_attributes_(device_attributes) { in Device()
Ddevice.h59 Device(Env* env, const DeviceAttributes& device_attributes);
/external/tensorflow/tensorflow/core/tpu/
Dvirtual_device.cc69 const DeviceAttributes& device_attributes) in VirtualDevice() argument
70 : Device(env, device_attributes) {} in VirtualDevice()
Dvirtual_device.h27 VirtualDevice(Env* env, const DeviceAttributes& device_attributes);
/external/tensorflow/tensorflow/compiler/jit/
Dincrease_dynamism_for_auto_jit_pass_test.cc44 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
45 : Device(nullptr, device_attributes) {} in FakeDevice()
52 DeviceAttributes device_attributes; in Make() local
53 device_attributes.set_name(name); in Make()
54 device_attributes.set_device_type(DeviceType(type).type()); in Make()
55 return absl::make_unique<FakeDevice>(device_attributes); in Make()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dmlir_tf2xla.cc49 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
50 : Device(nullptr, device_attributes) {} in FakeDevice()
/external/tensorflow/tensorflow/core/distributed_runtime/eager/
Deager_service_impl.cc307 std::vector<DeviceAttributes> device_attributes; in CreateContext() local
308 device_mgr->ListDeviceAttributes(&device_attributes); in CreateContext()
310 for (const auto& da : device_attributes) { in CreateContext()
400 std::vector<DeviceAttributes> device_attributes; in UpdateContext() local
401 device_mgr->ListDeviceAttributes(&device_attributes); in UpdateContext()
403 for (const auto& da : device_attributes) { in UpdateContext()
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto6 import "tensorflow/core/framework/device_attributes.proto";
127 repeated DeviceAttributes device_attributes = 2; field
153 repeated DeviceAttributes device_attributes = 1; field
Dworker.proto22 import "tensorflow/core/framework/device_attributes.proto";
49 repeated DeviceAttributes device_attributes = 1; field
549 DeviceAttributes device_attributes = 6; field
561 repeated DeviceAttributes device_attributes = 8; field

12