Searched refs:CompositeDevice (Results 1 – 12 of 12) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | composite_device_test.cc | 29 std::unique_ptr<CompositeDevice> composite_device = in TEST()
30 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
45 std::unique_ptr<CompositeDevice> composite_device = in TEST()
46 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/0, in TEST()
57 std::unique_ptr<CompositeDevice> composite_device = in TEST()
58 CompositeDevice::MakeDevice(underlying_devices, /*unique_device_id=*/1, in TEST()
75 std::unique_ptr<CompositeDevice> composite_device = in TEST()
76 CompositeDevice::MakeDevice(underlying_devices, composite_device_name, in TEST()
|
| D | composite_device.cc | 25 std::unique_ptr<CompositeDevice> CompositeDevice::MakeDevice( in MakeDevice()
32 return CompositeDevice::MakeDevice(underlying_devices, device_name, status); in MakeDevice()
35 std::unique_ptr<CompositeDevice> CompositeDevice::MakeDevice( in MakeDevice()
73 new CompositeDevice(device_attributes, underlying_devices)); in MakeDevice()
|
| D | composite_device.h | 32 class CompositeDevice : public Device {
47 static std::unique_ptr<CompositeDevice> MakeDevice(
52 static std::unique_ptr<CompositeDevice> MakeDevice(
59 CompositeDevice(const DeviceAttributes& device_attributes, in CompositeDevice() function
66 TF_DISALLOW_COPY_AND_ASSIGN(CompositeDevice);
|
| D | process_function_library_runtime.h | 221 void AddCompositeDevice(CompositeDevice* d) TF_LOCKS_EXCLUDED(mu_) { in AddCompositeDevice()
460 std::vector<CompositeDevice*> composite_devices_ TF_GUARDED_BY(mu_);
|
| D | process_function_library_runtime_test.cc | 163 void AddCompositeDevice(CompositeDevice* d) { in AddCompositeDevice()
860 std::unique_ptr<CompositeDevice> composite_device = in TEST_F()
861 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
1196 std::unique_ptr<CompositeDevice> composite_device = in TEST_F()
1197 CompositeDevice::MakeDevice({device0_->name(), device1_->name()}, in TEST_F()
|
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | context_test.cc | 85 TEST_F(EagerContextTest, CompositeDevice) { in TEST_F() argument
90 CompositeDevice* composite_device_0 = nullptr; in TEST_F()
96 CompositeDevice* device = nullptr; in TEST_F()
100 CompositeDevice* composite_device_1 = nullptr; in TEST_F()
106 CompositeDevice* composite_device_2 = nullptr; in TEST_F()
128 CompositeDevice* composite_device_0 = nullptr; in TEST_F()
133 CompositeDevice* device = nullptr; in TEST_F()
142 CompositeDevice* composite_device_1 = nullptr; in TEST_F()
|
| D | execute_node_test.cc | 62 std::unique_ptr<CompositeDevice> composite_device = in TEST()
63 CompositeDevice::MakeDevice({device0->name(), device1->name()}, in TEST()
|
| D | tensor_handle_test.cc | 192 CompositeDevice* device = in TEST_F()
193 reinterpret_cast<CompositeDevice*>(packed_handle->device()); in TEST_F()
239 CompositeDevice* device = in TEST_F()
240 reinterpret_cast<CompositeDevice*>(packed_handle->device()); in TEST_F()
|
| D | context.h | 477 CompositeDevice** device) const;
490 CompositeDevice** composite_device);
600 absl::flat_hash_map<uint64, std::unique_ptr<CompositeDevice>>
|
| D | context.cc | 896 StringPiece device_name, CompositeDevice** device) const { in FindCompositeDeviceFromName()
920 CompositeDevice** composite_device) { in FindOrCreateCompositeDevice()
936 std::unique_ptr<CompositeDevice> device; in FindOrCreateCompositeDevice()
940 device = CompositeDevice::MakeDevice(underlying_devices, in FindOrCreateCompositeDevice()
944 device = CompositeDevice::MakeDevice(underlying_devices, device_name, &s); in FindOrCreateCompositeDevice()
|
| D | tensor_handle.cc | 323 CompositeDevice* composite_device = nullptr; in CreatePackedHandle()
|
| D | execute.cc | 430 CompositeDevice* composite_device = nullptr; in GetOrCreateKernelAndDevice()
|