- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 49 for pack (0.17 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TF_Status* status); // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle. template <std::size_t num_replicas> void ExtractPerDeviceValues( TFE_Context* context, TFE_TensorHandle* input, std::array<TensorHandlePtr, num_replicas>* components, TF_Status* status); // Helper to pack `num_replicas` TFE_TensorHandles into one parallel handle. template <std::size_t num_replicas>
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
->DebugString(), " as input to a parallel operation. First pack non-parallel " "tensors for each device into a parallel tensor explicitly.") .c_str()); return absl::nullopt; } } else { parallel_inputs.push_back(absl::get<ParallelTensor*>(input)); } } absl::optional<std::vector<std::unique_ptr<ParallelTensor>>>
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) -
tensorflow/c/eager/parallel_device/parallel_device_test.cc
Multiply(context.get(), second_combined_value.get(), second_negative_one.get(), status.get())); ASSERT_EQ(TF_GetCode(status.get()), TF_OK) << TF_Message(status.get()); // Un-pack the parallel tensor to verify that the operation was // successful. The resulting structure should be: // second_device{first_device{1. * 3., 2. * 3.}, 3. * 3.}. std::array<TensorHandlePtr, 2> second_components;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 29.3K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
&FloatDeallocator, nullptr), TF_DeleteTensor); return TensorHandlePtr(TFE_NewTensorHandle(tensor.get(), status)); } // Helper to un-pack `num_replicas` TFE_TensorHandles from one parallel handle. template <std::size_t num_replicas> void ExtractPerDeviceValues( TFE_Context* context, TFE_TensorHandle* input,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jun 15 15:44:44 GMT 2021 - 12.5K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
components.reserve(underlying_devices_.size()); for (int j = 0; j < underlying_devices_.size(); ++j) { components.push_back(std::move(per_device_output_tensors[j][i])); } 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()),
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/c_api_distributed_test.cc
// Add a sync point to make sure that variables have been initialized // before the function execution starts. TFE_ContextAsyncWait(ctx, status); EXPECT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); // Pack 3 variable handles into one TFE_TensorHandle. // When remote is false, function device is placed on task0. Handle types are // REMOTE, REMOTE, LOCAL on task0. When remote is true, function device is
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 23.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
TFE_DeleteTensorHandle(outputs[i]); } } return status.status; } tensorflow::Status Pack(absl::Span<ImmediateExecutionTensorHandle*> handles, ImmediateExecutionTensorHandle** result) override { TF_Status status; *result = tensorflow::unwrap(device_.pack(context_, tensorflow::wrap(handles.data()),
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Mar 12 20:00:09 GMT 2024 - 43.9K bytes - Viewed (2) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
// fetcher calls that the cache makes. calls.push_back(0); // Cache miss - drains an element from `calls`. TF_EXPECT_OK(ReadCache(&cache, "", 0, 1, &out)); // Cache hit - does not drain an element from `calls`. TF_EXPECT_OK(ReadCache(&cache, "", 0, 1, &out)); calls.push_back(block_size); // Cache miss followed by cache hit.
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Fri Oct 15 03:16:57 GMT 2021 - 23.2K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
TF_RETURN_IF_ERROR(parent_op_->AddInput(input)); forward_op_.inputs.push_back(input); return OkStatus(); } Status TapeOperation::AddInputList( absl::Span<AbstractTensorHandle* const> inputs) { TF_RETURN_IF_ERROR(parent_op_->AddInputList(inputs)); for (auto input : inputs) { forward_op_.inputs.push_back(input); } return OkStatus(); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
.github/workflows/issue-on-pr-rollback.yml
# See the License for the specific language governing permissions and # limitations under the License. # ============================================================================== name: Creates a GitHub Issue when a PR Rolled back via Commit to Master on: push: branches: - master permissions: {} jobs: create-issue-on-pr-rollback: runs-on: ubuntu-latest permissions: contents: read
Others - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Sep 12 16:40:29 GMT 2023 - 1.5K bytes - Viewed (0)