Home
last modified time | relevance | path

Searched refs:ParallelTensor (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_lib.h48 class ParallelTensor; variable
67 std::unique_ptr<ParallelTensor> CopyToParallelDevice(TFE_Context* context,
73 std::unique_ptr<ParallelTensor> ScalarsFromSequence(
78 std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context,
95 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> Execute(
96 TFE_Context* context, const std::vector<ParallelTensor*>& inputs,
113 const std::vector<ParallelTensor*>& inputs,
129 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> Join(
158 class ParallelTensor {
163 static std::unique_ptr<ParallelTensor> FromTensorHandles(
[all …]
Dparallel_device.cc42 absl::variant<std::unique_ptr<ParallelTensor>, TensorHandlePtr>;
45 absl::variant<ParallelTensor*, TFE_TensorHandle*>;
87 if (absl::holds_alternative<ParallelTensor*>(inputs[i])) { in ExecuteWithSpecialOps()
102 result_content.push_back(ParallelTensor::FromTensorHandles( in ExecuteWithSpecialOps()
129 ParallelTensor* t = absl::get<ParallelTensor*>(inputs[0]); in ExecuteWithSpecialOps()
141 std::vector<ParallelTensor*> parallel_inputs; in ExecuteWithSpecialOps()
142 std::vector<std::unique_ptr<ParallelTensor>> implicitly_broadcast_tensors; in ExecuteWithSpecialOps()
154 std::unique_ptr<ParallelTensor> parallel_tensor( in ExecuteWithSpecialOps()
161 parallel_inputs.push_back(absl::get<ParallelTensor*>(input)); in ExecuteWithSpecialOps()
164 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> in ExecuteWithSpecialOps()
[all …]
Dparallel_device_lib.cc272 std::unique_ptr<ParallelTensor> ParallelDevice::CopyToParallelDevice( in CopyToParallelDevice()
282 return ParallelTensor::FromTensorHandles(*this, std::move(components), in CopyToParallelDevice()
286 std::unique_ptr<ParallelTensor> ParallelDevice::DeviceIDs( in DeviceIDs()
296 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
298 const std::vector<ParallelTensor*>& inputs, in Execute()
321 TFE_Context* context, const std::vector<ParallelTensor*>& inputs, in StartExecute()
339 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
343 absl::optional<std::vector<std::unique_ptr<ParallelTensor>>> result; in Join()
384 std::vector<std::unique_ptr<ParallelTensor>> per_device_outputs; in Join()
393 per_device_outputs.push_back(ParallelTensor::FromTensorHandles( in Join()
[all …]
Dparallel_device_lib_test.cc63 parallel_device.Execute(context.get(), std::vector<ParallelTensor*>(), in TEST()
67 const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs; in TEST()
68 std::vector<ParallelTensor*> handle_inputs; in TEST()
84 parallel_device.Execute(context.get(), std::vector<ParallelTensor*>(), in TEST()
119 parallel_device.StartExecute(context.get(), std::vector<ParallelTensor*>(), in TEST()
126 const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs; in TEST()
187 std::unique_ptr<ParallelTensor> reduced_values = in TEST()
191 std::unique_ptr<ParallelTensor> run_collective = in TEST()
201 ParallelTensor* parallel_result = (*outputs)[0].get(); in TEST()
251 std::unique_ptr<ParallelTensor> unknown_length_vector = in TEST()
[all …]