HomeSort by relevance Sort by last modified time
    Searched refs:device_attributes (Results 1 - 8 of 8) sorted by null

  /external/tensorflow/tensorflow/core/common_runtime/
device.cc 26 Device::Device(Env* env, const DeviceAttributes& device_attributes)
27 : DeviceBase(env), device_attributes_(device_attributes) {
device.h 37 #include "tensorflow/core/framework/device_attributes.pb_text.h"
38 #include "tensorflow/core/framework/device_attributes.pb.h"
56 Device(Env* env, const DeviceAttributes& device_attributes);
placer_test.cc 26 #include "tensorflow/core/framework/device_attributes.pb.h"
68 explicit FakeDevice(const DeviceAttributes& device_attributes)
69 : Device(nullptr, device_attributes) {}
77 DeviceAttributes device_attributes; local
78 device_attributes.set_name(name);
79 device_attributes.set_device_type(DeviceType("FakeCPU").type());
80 return std::unique_ptr<Device>(new FakeDevice(device_attributes));
84 DeviceAttributes device_attributes; local
85 device_attributes.set_name(name);
86 device_attributes.set_device_type(DeviceType("FakeGPU").type())
    [all...]
  /external/tensorflow/tensorflow/core/distributed_runtime/
session_mgr_test.cc 27 explicit FakeDevice(const DeviceAttributes& device_attributes)
28 : Device(nullptr, device_attributes) {}
36 DeviceAttributes device_attributes; local
37 device_attributes.set_name(name);
38 device_attributes.set_device_type(DeviceType("FakeCPU").type());
39 return std::unique_ptr<Device>(new FakeDevice(device_attributes));
worker_cache_partial.cc 80 for (auto& dev_attr : resp.device_attributes()) {
remote_device.cc 91 for (const DeviceAttributes& da : call->resp.device_attributes()) {
  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_testlib.cc 72 std::vector<DeviceAttributes> device_attributes; local
  /external/tensorflow/tensorflow/core/protobuf/
worker.proto 27 import "tensorflow/core/framework/device_attributes.proto";
46 repeated DeviceAttributes device_attributes = 1;

Completed in 575 milliseconds