Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/tools/optimization/
Doptimization_pass_runner.cc52 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
53 : Device(nullptr, device_attributes) {} in FakeDevice()
66 DeviceAttributes device_attributes; in Make() local
67 device_attributes.set_name(name); in Make()
68 device_attributes.set_device_type(DeviceType(type).type()); in Make()
69 return std::unique_ptr<Device>(new 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()
Dworker_cache_partial.cc80 for (auto& dev_attr : resp.device_attributes()) { in RefreshDeviceStatus()
Dremote_device.cc91 for (const DeviceAttributes& da : call->resp.device_attributes()) { in NewRemoteDevices()
Ddevice_resolver_distributed.cc106 for (const DeviceAttributes& da : resp->device_attributes()) { in RefreshRemoteAttributes()
Ddevice_resolver_distributed_test.cc117 for (const auto& it : resp.device_attributes()) { in GetDeviceLocalityAsync()
Dcollective_rma_distributed_test.cc155 for (const auto& it : resp.device_attributes()) { in GetDeviceLocalityAsync()
Dcollective_param_resolver_distributed_test.cc115 for (const auto& it : resp.device_attributes()) { in GetDeviceLocalityAsync()
/external/tensorflow/tensorflow/core/common_runtime/
Ddevice.cc26 Device::Device(Env* env, const DeviceAttributes& device_attributes) in Device() argument
27 : DeviceBase(env), device_attributes_(device_attributes) { in Device()
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()
Dplacer_test.cc84 explicit FakeDevice(const DeviceAttributes& device_attributes) in FakeDevice() argument
85 : Device(nullptr, device_attributes) {} in FakeDevice()
93 DeviceAttributes device_attributes; in MakeCPU() local
94 device_attributes.set_name(name); in MakeCPU()
95 device_attributes.set_device_type(DeviceType("FakeCPU").type()); in MakeCPU()
96 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeCPU()
100 DeviceAttributes device_attributes; in MakeGPU() local
101 device_attributes.set_name(name); in MakeGPU()
102 device_attributes.set_device_type(DeviceType("FakeGPU").type()); in MakeGPU()
103 return std::unique_ptr<Device>(new FakeDevice(device_attributes)); in MakeGPU()
Ddevice.h62 Device(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/core/distributed_runtime/rpc/
Dgrpc_testlib.cc78 std::vector<DeviceAttributes> device_attributes; in MakeTestCluster() local
/external/tensorflow/tensorflow/contrib/makefile/
Dtf_pb_text_files.txt7 tensorflow/core/framework/device_attributes.pb_text.cc
Dproto_text_pb_cc_files.txt7 tensorflow/core/framework/device_attributes.pb.cc
Dtf_proto_files.txt11 tensorflow/core/framework/device_attributes.proto
Dproto_text_pb_h_files.txt7 tensorflow/core/framework/device_attributes.pb.h
/external/tensorflow/tensorflow/core/protobuf/
Deager_service.proto6 import "tensorflow/core/framework/device_attributes.proto";
83 repeated DeviceAttributes device_attributes = 2; field
Dworker.proto27 import "tensorflow/core/framework/device_attributes.proto";
48 repeated DeviceAttributes device_attributes = 1; field
Dmaster.proto26 import "tensorflow/core/framework/device_attributes.proto";
/external/tensorflow/tensorflow/contrib/cmake/
Dtf_core_framework.cmake146 "tensorflow/core/framework/device_attributes.proto"
/external/skia/third_party/skcms/
Dskcms.cc167 uint8_t device_attributes [ 8]; member
/external/skqp/third_party/skcms/
Dskcms.cc167 uint8_t device_attributes [ 8]; member
/external/tensorflow/tensorflow/core/
DBUILD194 "framework/device_attributes.proto",