Home
last modified time | relevance | path

Searched refs:TensorHandlePtr (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/c/eager/parallel_device/
Dparallel_device_test.cc128 TensorHandlePtr cpu_value(FloatTensorHandle(3., status.get())); in TEST()
132 TensorHandlePtr device_value(TFE_TensorHandleCopyToDevice( in TEST()
142 std::array<TensorHandlePtr, 2> components; in TEST()
161 TensorHandlePtr copy_back(TFE_TensorHandleCopyToDevice( in TEST()
194 TensorHandlePtr size_two( in TEST()
197 TensorHandlePtr size_three( in TEST()
203 TensorHandlePtr combined_value = CreatePerDeviceValues( in TEST()
250 TensorHandlePtr value_one(FloatTensorHandle(1., status.get())); in TEST()
251 TensorHandlePtr value_two(FloatTensorHandle(2., status.get())); in TEST()
254 TensorHandlePtr first_combined_value = CreatePerDeviceValues( in TEST()
[all …]
Dparallel_device_testlib.cc78 TensorHandlePtr Variable::Read(TFE_Context* context, TF_Status* status) { in Read()
93 return TensorHandlePtr(var_value); in Read()
132 TensorHandlePtr FloatTensorHandle(float v, TF_Status* status) { in FloatTensorHandle()
140 return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); in FloatTensorHandle()
144 TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v, in VectorFloatTensorHandle()
154 return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); in VectorFloatTensorHandle()
161 std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status) { in ExtractPerDeviceValues() argument
182 TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first, in Multiply()
199 return TensorHandlePtr(result_handle); in Multiply()
231 TensorHandlePtr initial_value = FloatTensorHandle(20., status.get()); in BasicTestsForTwoDevices()
[all …]
Dparallel_device_testlib.h55 TensorHandlePtr Read(TFE_Context* context, TF_Status* status);
77 TensorHandlePtr FloatTensorHandle(float v, TF_Status* status);
80 TensorHandlePtr VectorFloatTensorHandle(const std::vector<float>& v,
87 std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status);
91 TensorHandlePtr CreatePerDeviceValues(
96 TensorHandlePtr Multiply(TFE_Context* context, TFE_TensorHandle* first,
117 TensorHandlePtr CreatePerDeviceValues( in CreatePerDeviceValues()
136 return TensorHandlePtr(result_handle); in CreatePerDeviceValues()
Dparallel_device_lib.cc94 std::vector<TensorHandlePtr> Join(TF_Status* status);
102 std::vector<TensorHandlePtr>* outputs, TF_Status* status) const
133 std::vector<TensorHandlePtr> op_outputs_ TF_GUARDED_BY(execution_mutex_);
165 op_outputs_ = std::vector<TensorHandlePtr>(); in Run()
199 std::vector<TensorHandlePtr> DeviceThread::Join(TF_Status* status) { in Join()
200 std::vector<TensorHandlePtr> result; in Join()
225 std::vector<TensorHandlePtr>* outputs, in Execute()
274 std::vector<TensorHandlePtr> components; in CopyToParallelDevice()
345 std::vector<std::vector<TensorHandlePtr>> per_device_output_tensors; in Join()
387 std::vector<TensorHandlePtr> components; in Join()
[all …]
Dparallel_device_lib.h46 using TensorHandlePtr = std::unique_ptr<TFE_TensorHandle, TensorHandleDeleter>; variable
165 std::vector<TensorHandlePtr> components, TF_Status* status);
170 std::vector<TensorHandlePtr> components, absl::Span<const int64> shape,
187 std::vector<TensorHandlePtr> tensors, in ParallelTensor()
194 std::vector<TensorHandlePtr> tensors, const TF_DataType dtype) in ParallelTensor()
201 const std::vector<TensorHandlePtr> tensors_;
213 std::vector<TensorHandlePtr> components; in ScalarsFromSequence()
Dparallel_device_remote_test.cc123 TensorHandlePtr value_one(FloatTensorHandle(3., status.get())); in TEST()
125 TensorHandlePtr value_two(FloatTensorHandle(-2., status.get())); in TEST()
129 TensorHandlePtr combined_value = CreatePerDeviceValues( in TEST()
135 TensorHandlePtr multiply_result( in TEST()
139 std::array<TensorHandlePtr, 2> out_components; in TEST()
Dparallel_device.cc42 absl::variant<std::unique_ptr<ParallelTensor>, TensorHandlePtr>;
84 std::vector<TensorHandlePtr> components; in ExecuteWithSpecialOps()
133 TensorHandlePtr this_output( in ExecuteWithSpecialOps()
213 TensorHandlePtr ParallelTensorToTensorHandle( in ParallelTensorToTensorHandle()
220 return TensorHandlePtr(TFE_NewCustomDeviceTensorHandle( in ParallelTensorToTensorHandle()
338 if (absl::holds_alternative<TensorHandlePtr>(typed_output)) { in ParallelDeviceExecute()
339 outputs[i] = absl::get<TensorHandlePtr>(typed_output).release(); in ParallelDeviceExecute()
Dparallel_device_lib_test.cc241 TensorHandlePtr two_vector = VectorFloatTensorHandle({3., 4.}, status.get()); in TEST()
243 TensorHandlePtr three_vector = in TEST()
247 std::vector<TensorHandlePtr> vector_handles; in TEST()
259 TensorHandlePtr scalar = FloatTensorHandle(2., status.get()); in TEST()
263 std::vector<TensorHandlePtr> mixed_handles; in TEST()