- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for FromTensorHandles (0.31 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.h
static std::unique_ptr<ParallelTensor> FromTensorHandles( const ParallelDevice& parallel_device, std::vector<TensorHandlePtr> components, TF_Status* status); // Uses the provided shape without additional checks, which avoids blocking // when ParallelTensor::Shape is called. static std::unique_ptr<ParallelTensor> FromTensorHandles( const ParallelDevice& parallel_device,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
if (expected_output_shapes[i].IsFullyDefined()) { per_device_outputs.push_back(ParallelTensor::FromTensorHandles( *this, std::move(components), absl::Span<const int64_t>(expected_output_shapes[i].dim_sizes()), status)); } else { per_device_outputs.push_back(ParallelTensor::FromTensorHandles( *this, std::move(components), status)); }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Feb 09 07:47:20 GMT 2024 - 25.4K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
vector_handles.reserve(2); vector_handles.push_back(std::move(two_vector)); vector_handles.push_back(std::move(three_vector)); std::unique_ptr<ParallelTensor> unknown_length_vector = ParallelTensor::FromTensorHandles( parallel_device, std::move(vector_handles), status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); const std::vector<int64_t>* shape;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
absl::get<TFE_TensorHandle*>(inputs[i]), status)); } std::vector<MaybeParallelTensorOwned> result_content; result_content.reserve(1); result_content.push_back(ParallelTensor::FromTensorHandles( parallel_device, std::move(components), status)); if (TF_GetCode(status) != TF_OK) return result; result.emplace(std::move(result_content)); return result;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0)