| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | device_resolver_distributed_test.cc | 37 std::unique_ptr<Device> NewDevice(const string& type, const string& name) { in NewDevice() function
56 NewDevice("CPU", "/job:worker/replica:0/task:0/device:CPU:0")); in SetUp()
58 NewDevice("CPU", "/job:worker/replica:0/task:0/device:CPU:1")); in SetUp()
65 NewDevice("CPU", "/job:worker/replica:0/task:1/device:CPU:0") in SetUp()
68 NewDevice("CPU", "/job:worker/replica:0/task:1/device:CPU:1") in SetUp()
119 NewDevice("CPU", "/job:worker/replica:0/task:2/device:CPU:0") in TEST_F()
122 NewDevice("CPU", "/job:worker/replica:0/task:2/device:CPU:1") in TEST_F()
155 NewDevice("CPU", "/job:worker/replica:0/task:0/device:CPU:0") in TEST_F()
158 NewDevice("CPU", "/job:worker/replica:0/task:0/device:CPU:1") in TEST_F()
|
| D | collective_param_resolver_distributed_test.cc | 34 static std::unique_ptr<Device> NewDevice(const string& type, in NewDevice() function
155 devices.push_back(NewDevice( in DefineWorker()
|
| D | collective_rma_distributed_test.cc | 45 static std::unique_ptr<Device> NewDevice(const string& type, in NewDevice() function
244 devices.push_back(NewDevice( in DefineWorker()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | remote_mgr_test.cc | 46 DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0")); in RemoteMgrTest()
49 DeviceFactory::NewDevice("CPU", {}, "/job:worker/replica:0/task:0")); in RemoteMgrTest()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | execute_node_test.cc | 52 DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0")); in TEST()
57 DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:1")); 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 | 47 DeviceFactory::NewDevice("CPU", {}, "/job:a/replica:0/task:0"));
|
| D | mkl_eager_op_rewrite_test.cc | 38 absl::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in CreateOp()
|
| /external/tensorflow/tensorflow/core/kernels/mlir_generated/ |
| D | cpu_binary_ops_test.cc | 30 tensorflow::DeviceFactory::NewDevice("CPU", {}, in SetUp()
|
| D | cpu_unary_ops_test.cc | 30 tensorflow::DeviceFactory::NewDevice("CPU", {}, in SetUp()
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | device_factory.h | 48 static std::unique_ptr<Device> NewDevice(const string& type,
|
| D | device_factory.cc | 188 std::unique_ptr<Device> DeviceFactory::NewDevice(const string& type, in NewDevice() function in tensorflow::DeviceFactory
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | depthwise_conv_ops_test.cc | 44 std::unique_ptr<tensorflow::Device>(DeviceFactory::NewDevice( in Run()
|
| D | constant_op_test.cc | 60 std::unique_ptr<Device> device(DeviceFactory::NewDevice( in PersistentMemoryTrackingTest()
|
| D | restore_op_test.cc | 86 DeviceFactory::NewDevice("CPU", {}, "/job:a/replica:0/task:0")); in TEST_F()
390 DeviceFactory::NewDevice("CPU", {}, "/job:a/replica:0/task:0")); in TEST_F()
|
| D | segment_reduction_ops_test.cc | 46 DeviceFactory::NewDevice("CPU", {}, "/job:a/replica:0/task:0")); in BM_SegmentReduction()
|
| D | parse_tensor_test.cc | 49 DeviceFactory::NewDevice("CPU", {}, "/job:a/replica:0/task:0")); in ParseSerializedWithNodeDef()
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | resize_nearest_neighbor_op_test.cc | 45 DeviceFactory::NewDevice(/*type=*/"GPU", /*options=*/{}, in SetUp()
84 DeviceFactory::NewDevice(/*type=*/"GPU", /*options=*/{}, in SetUp()
|
| /external/tensorflow/tensorflow/compiler/tests/ |
| D | unary_ops_composition_test.cc | 60 std::unique_ptr<tensorflow::Device>(DeviceFactory::NewDevice( in RunComposedOp()
|
| /external/tensorflow/tensorflow/c/experimental/saved_model/core/ |
| D | test_utils.cc | 41 DeviceFactory::NewDevice("CPU", {}, "/job:localhost/replica:0/task:0")); in CreateTestingDeviceMgr()
|
| /external/tensorflow/tensorflow/core/data/ |
| D | standalone.cc | 51 auto device_mgr = absl::make_unique<StaticDeviceMgr>(DeviceFactory::NewDevice( in FromGraph()
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | buf_rendezvous_test.cc | 31 static std::unique_ptr<Device> NewDevice(const string& name, in NewDevice() function in tensorflow::__anon5d7062b30111::BufRendezvousTest
51 devices.push_back(NewDevice(device, type, incarnation)); in InitializeDevice()
|
| /external/tensorflow/tensorflow/compiler/tf2tensorrt/kernels/ |
| D | trt_engine_resource_ops_test.cc | 108 DeviceFactory::NewDevice("GPU", {}, "/job:worker/replica:0/task:0")); in TEST_F()
|