Searched refs:ParallelDevice (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib.h | 53 class ParallelDevice {
57 explicit ParallelDevice(const std::vector<std::string>& devices,
60 ~ParallelDevice();
164 const ParallelDevice& parallel_device,
169 const ParallelDevice& parallel_device,
186 ParallelTensor(const ParallelDevice& device, in ParallelTensor()
193 ParallelTensor(const ParallelDevice& device, in ParallelTensor()
200 const ParallelDevice& device_;
210 std::unique_ptr<ParallelTensor> ParallelDevice::ScalarsFromSequence( in ScalarsFromSequence()
|
| D | parallel_device_lib.cc | 258 ParallelDevice::ParallelDevice(const std::vector<std::string>& devices, in ParallelDevice() function in tensorflow::parallel_device::ParallelDevice
270 ParallelDevice::~ParallelDevice() = default;
272 std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( in CopyToParallelDevice()
286 std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs( in DeviceIDs()
297 ParallelDevice::Execute(TFE_Context* context, in Execute()
320 void ParallelDevice::StartExecute( in StartExecute()
340 ParallelDevice::Join( in Join()
408 const ParallelDevice& parallel_device, in FromTensorHandles()
427 const ParallelDevice& parallel_device, in FromTensorHandles()
|
| D | parallel_device.cc | 54 std::unique_ptr<ParallelDevice> parallel_device) in NamedParallelDevice()
57 const ParallelDevice& device() const { return *parallel_device_; } in device()
61 std::unique_ptr<ParallelDevice> parallel_device_;
65 const ParallelDevice& parallel_device, in ExecuteWithSpecialOps()
241 const ParallelDevice& dev = named_device->device(); in CopyToParallelDevice()
378 std::unique_ptr<ParallelDevice> parallel_device( in AllocateParallelDevice()
379 new ParallelDevice(underlying_devices_vector)); in AllocateParallelDevice()
|
| D | parallel_device_lib_test.cc | 54 ParallelDevice parallel_device(std::move(devices)); in TEST()
110 ParallelDevice parallel_device(std::move(devices)); in TEST()
152 ParallelDevice parallel_device(devices); in TEST()
240 ParallelDevice parallel_device(std::move(devices)); in TEST()
305 ParallelDevice parallel_device(std::move(devices)); in TEST()
|
| /external/tensorflow/tensorflow/python/distribute/parallel_device/ |
| D | parallel_device.py | 53 class ParallelDevice(object): class
|
| D | parallel_device_test.py | 117 self.device = parallel_device.ParallelDevice(components=[
|