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

  /external/tensorflow/tensorflow/core/distributed_runtime/rpc/
grpc_testlib.cc 72 std::vector<DeviceAttributes> device_attributes; local
  /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));
  /external/tensorflow/tensorflow/core/common_runtime/
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...]

Completed in 57 milliseconds