- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for expected_output_shapes (0.18 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
TF_Status* status) const { std::vector<PartialTensorShape> expected_output_shapes(expected_max_outputs); StartExecute(context, inputs, operation_name, attributes, expected_max_outputs, *default_cancellation_manager_); auto result = Join(expected_output_shapes, status); if (TF_GetCode(status) != TF_OK) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> await_status(
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 25.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.h
// run and have results. If is_async=true it means that all of the // device-specific executors have scheduled the op. // // Accepts inferred shapes for outputs (`expected_output_shapes`), which if // fully defined will avoid querying the shapes of the underlying // TensorHandles when ParallelTensor::Shape is called. This allows async // computation to continue without blocking. //
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 13.1K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib_test.cc
/*expected_max_outputs=*/1, cancellation_manager); auto outputs = parallel_device.Join( /*expected_output_shapes=*/{PartialTensorShape({})}, status.get()); ASSERT_TRUE(TF_GetCode(status.get()) == TF_OK) << TF_Message(status.get()); const std::vector<std::unique_ptr<ParallelTensor>>& handles = *outputs;
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 15.6K bytes - Viewed (0)