- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 28 for shouldn (0.32 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_testlib.cc
TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status); TFE_OpSetAttrString(op.get(), "container", "", 0); // Use the special GUID for no buffer sharing // // TODO(allenl): Should we provide a better API for this? AFAIK this is the // only reasonable way to make variables with no aliasing using the eager C // API. std::string no_sharing = "cd2c89b7-88b7-44c8-ad83-06c2a9158347";
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.cc
const TFE_OpAttrs* attributes, int expected_max_outputs, TF_Status* status) { absl::optional<std::vector<MaybeParallelTensorOwned>> result; // TODO(allenl): We should remove "TPU" from these op names at the very least, // or consider other ways of packing/unpacking parallel tensors. if (operation_name == std::string("TPUReplicatedInput")) {
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_lib.cc
// // DeviceThread itself is thread-safe, in that StartExecute will block if there // is a pending execution. Since StartExecute is equivalent to grabbing a lock, // multiple DeviceThreads should always be accessed in the same order to avoid // deadlocks. class DeviceThread { public: // Starts a background thread waiting for `StartExecute`. explicit DeviceThread(const std::string& device, const bool is_async,
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_test.cc
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; ExtractPerDeviceValues(context.get(), multiply_result.get(),
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/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc
// We will work with content_view instead of content. static const absl::string_view content_view = content; namespace gcs = google::cloud::storage; static std::string InitializeTmpDir() { // This env should be something like `gs://bucket/path` const char* test_dir = getenv("GCS_TEST_TMPDIR"); if (test_dir != nullptr) { std::string bucket, object; TF_Status* status = TF_NewStatus();
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Aug 31 12:04:23 GMT 2020 - 24.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test_util.cc
EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); auto remote_arg = tensorflow::TensorHandleFromInterface(tensorflow::unwrap(h1_task2)); // The input handles should never change since they have been mirrored. ASSERT_FALSE(remote_arg->HasLocalMirror(nullptr)); } TFE_TensorHandle* retvals[1]; int num_retvals = 1;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Dec 11 22:56:03 GMT 2020 - 9.1K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker_test.cc
absl::Span<AbstractTensorHandle*> outputs) { return ops::Mul(ctx, inputs[0], inputs[1], &outputs[0], "Mul"); } // TODO(vnvo2409): Add more tests from `python/ops/gradient_checker_v2_test.py`. // These tests should not be confused with `[*]_grad_test` which compare the // result of `gradient_checker` and `[*]_grad`. The tests here test the // functionality of `gradient_checker` by comparing the result with expected
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 6.5K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
#endif // !IS_MOBILE_PLATFORM return tensorflow::wrap(eager_context); } void TFE_DeleteContext(TFE_Context* ctx) { if (ctx == nullptr) { return; } // ctx->RefCountIsOne() should be true here. tensorflow::unwrap(ctx)->Release(); } TF_DeviceList* TFE_ContextListDevices(TFE_Context* ctx, TF_Status* status) { TF_DeviceList* l = new TF_DeviceList;
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/eager/c_api_experimental.cc
tensorflow::unwrap(op_to_reset); op->Clear(); status->status = op->Reset(op_or_function_name, raw_device_name); } else { TF_SetStatus(status, TF_INVALID_ARGUMENT, "op_to_reset should not be nullptr"); } } void TFE_ContextEnableGraphCollection(TFE_Context* ctx) { tensorflow::unwrap(ctx)->SetShouldStoreGraphs(true); } void TFE_ContextDisableGraphCollection(TFE_Context* ctx) {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 11 23:52:39 GMT 2024 - 35.9K bytes - Viewed (3) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
TF_SetStatusFromGCSStatus(metadata.status(), status); if (TF_GetCode(status) == TF_FAILED_PRECONDITION) TF_SetStatus(status, TF_ALREADY_EXISTS, path); } // TODO(vnvo2409): `RecursivelyCreateDir` should use `CreateDir` instead of the // default implementation. Because we could create an empty object whose // key is equal to the `path` and Google Cloud Console will automatically // display it as a directory tree.
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 23 06:55:53 GMT 2023 - 46.9K bytes - Viewed (0)