| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | device_mgr.cc | 31 StaticDeviceMgr::StaticDeviceMgr(std::vector<std::unique_ptr<Device>> devices) in StaticDeviceMgr() function in tensorflow::StaticDeviceMgr
55 StaticDeviceMgr::StaticDeviceMgr(std::unique_ptr<Device> device) in StaticDeviceMgr() function in tensorflow::StaticDeviceMgr
56 : StaticDeviceMgr([&device] { in StaticDeviceMgr()
62 StaticDeviceMgr::~StaticDeviceMgr() { in ~StaticDeviceMgr()
70 StringPiece StaticDeviceMgr::CopyToBackingStore(StringPiece s) { in CopyToBackingStore()
77 void StaticDeviceMgr::ListDeviceAttributes( in ListDeviceAttributes()
85 std::vector<Device*> StaticDeviceMgr::ListDevices() const { in ListDevices()
93 string StaticDeviceMgr::DebugString() const { in DebugString()
101 string StaticDeviceMgr::DeviceMappingString() const { in DeviceMappingString()
112 Status StaticDeviceMgr::LookupDevice(StringPiece name, Device** device) const { in LookupDevice()
[all …]
|
| D | device_mgr_test.cc | 43 TEST(StaticDeviceMgr, NoCPUDevice) { in TEST() argument
49 StaticDeviceMgr lm(std::move(devices)); in TEST()
53 TEST(StaticDeviceMgr, SomeCPUDevice) { in TEST() argument
62 StaticDeviceMgr lm(std::move(devices)); in TEST()
|
| D | device_mgr.h | 79 class StaticDeviceMgr : public DeviceMgr {
82 explicit StaticDeviceMgr(std::vector<std::unique_ptr<Device>> devices);
85 explicit StaticDeviceMgr(std::unique_ptr<Device> device);
87 ~StaticDeviceMgr() override;
111 TF_DISALLOW_COPY_AND_ASSIGN(StaticDeviceMgr);
|
| D | kernel_benchmark_testlib.h | 41 class StaticDeviceMgr; variable
93 std::unique_ptr<StaticDeviceMgr> device_mgr_;
|
| D | device_resolver_local_test.cc | 39 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in DeviceResolverLocalTest()
|
| D | collective_executor_mgr_test.cc | 44 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in CollectiveExecutorMgrTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | tensor_handle_test.cc | 37 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST()
103 device_mgr_ = new StaticDeviceMgr(std::move(devices)); in PackedTensorHandleTest()
135 StaticDeviceMgr* device_mgr_;
254 StaticDeviceMgr local_device_mgr(std::move(d0)); in TEST()
275 StaticDeviceMgr new_device_mgr(std::move(d1)); in TEST()
286 StaticDeviceMgr local_device_mgr(std::move(d_local)); in TEST()
339 device_mgr_ = new StaticDeviceMgr(std::move(devices)); in RemoteTensorHandleTest()
366 StaticDeviceMgr* device_mgr_;
378 StaticDeviceMgr device_mgr(std::move(devices)); in TEST_F()
414 StaticDeviceMgr local_device_mgr(std::move(devices)); in TEST()
|
| D | eager_operation_test.cc | 26 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST()
|
| D | custom_device_test.cc | 87 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST()
112 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST()
|
| D | eager_op_rewrite_registry_test.cc | 46 StaticDeviceMgr device_mgr(DeviceFactory::NewDevice( in TEST()
|
| D | kernel_and_device_test.cc | 49 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices));
|
| D | execute_node_test.cc | 51 StaticDeviceMgr device_mgr( in TEST()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 39 std::unique_ptr<StaticDeviceMgr> CreateTestingDeviceMgr() { in CreateTestingDeviceMgr()
40 return std::make_unique<StaticDeviceMgr>( in CreateTestingDeviceMgr()
|
| D | saved_variable_loading_test.cc | 48 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in SavedVariableLoadingTest()
55 std::unique_ptr<StaticDeviceMgr> device_mgr_;
|
| D | test_utils.h | 36 std::unique_ptr<StaticDeviceMgr> CreateTestingDeviceMgr();
|
| D | constant_loading_test.cc | 46 std::unique_ptr<StaticDeviceMgr> device_mgr_;
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | delegate_data.cc | 42 absl::make_unique<tensorflow::StaticDeviceMgr>(std::move(devices)); in Prepare()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ops/ |
| D | variable_ops_test.cc | 49 std::unique_ptr<StaticDeviceMgr> device_mgr_;
|
| D | restore_ops_test.cc | 47 std::unique_ptr<StaticDeviceMgr> device_mgr_;
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ops_testutil.cc | 79 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(device)); in OpsTestBase()
108 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(device)); in SetDevice()
|
| /external/tensorflow/tensorflow/compiler/jit/ |
| D | xla_kernel_creator_test.cc | 76 device_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in Init()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 59 dev_mgr_ = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in SetUp()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | upgrade_graph.cc | 128 return new StaticDeviceMgr(std::move(devices)); in GetStaticDeviceMgr()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 51 auto device_mgr = absl::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in FromGraph()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 51 auto device_mgr = absl::make_unique<StaticDeviceMgr>(std::move(devices)); in RemoteMgrTest()
|