- Sort Score
- Result 10 results
- Languages All
Results 31 - 40 of 40 for Device (0.19 sec)
-
tensorflow/c/eager/c_api_test.cc
// Remote device on `worker1`. const char remote_device[] = "/job:localhost/replica:0/task:1/device:CPU:0"; // `ctx_0`, `ctx_1`, `ctx_2` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(), remote_device) != device_names.end()); } {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 20:50:20 GMT 2023 - 94.6K bytes - Viewed (1) -
tensorflow/c/eager/c_api_experimental_test.cc
// Remote device on `worker2`. const char remote_device[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // `ctx_0`, `ctx_1` contains `remote_device`. { const std::vector<std::string>& device_names = ListDeviceNames(ctx_0); ASSERT_TRUE(std::find(device_names.begin(), device_names.end(), remote_device) != device_names.end()); } {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 03 03:14:26 GMT 2023 - 31.5K bytes - Viewed (1) -
tensorflow/c/c_api_experimental_test.cc
TF_Status* status = TF_NewStatus(); string lib_path = tensorflow::GetDataDependencyFilepath(tensorflow::io::JoinPath( "tensorflow", "core", "common_runtime", "next_pluggable_device", "c", "test_next_pluggable_device_plugin.so")); TF_Library* lib = TF_LoadPluggableDeviceLibrary(lib_path.c_str(), status); TF_Code code = TF_GetCode(status); string status_msg(TF_Message(status)); TF_DeleteStatus(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Jan 17 22:27:52 GMT 2023 - 13.1K bytes - Viewed (1) -
tensorflow/c/c_api_test.cc
CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); LOG(INFO) << "Device " << i << " has name " << device_name << ", type " << device_type; if (string(device_type) == DEVICE_GPU) { return device_name; } } // No GPU device found. return ""; } string GPUDeviceName() { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status(
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 96.9K bytes - Viewed (3) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem_test.cc
TF_DeleteStatus(status); return ""; } TF_DeleteStatus(status); // We add a random value into `test_dir` to ensures that two consecutive // runs are unlikely to clash. std::random_device rd; std::mt19937 gen(rd()); std::uniform_int_distribution<> distribution; std::string rng_val = std::to_string(distribution(gen)); return tensorflow::io::JoinPath(std::string(test_dir), rng_val); } else {
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/unified_api_testutil.cc
output->Unref(); } TF_RETURN_IF_ERROR(ctx->RegisterFunction(func)); } AbstractOperationPtr fn_op(ctx->CreateOperation()); TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); for (auto input : inputs) { TF_RETURN_IF_ERROR(fn_op->AddInput(input)); } int retvals = outputs.size() - null_indices.size(); std::vector<AbstractTensorHandle*> fn_outputs(retvals);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 5.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.cc
// parent_op->Unref(); } Status TapeOperation::Reset(const char* op, const char* raw_device_name) { forward_op_.op_name = op; forward_op_.attrs.Reset(op); forward_op_.inputs.clear(); forward_op_.outputs.clear(); return parent_op_->Reset(op, raw_device_name); } const string& TapeOperation::Name() const { return parent_op_->Name(); } const string& TapeOperation::DeviceName() const {
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/c_api_function_test.cc
TF_OperationDescription* desc = TF_NewOperation(host_graph_, func_name_, func_node_name_); for (auto input : inputs) { TF_AddInput(desc, input); } // Set device to CPU because some ops inside the function might not be // available on GPU. TF_SetDevice(desc, "/cpu:0"); *op = TF_FinishOperation(desc, s_); ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 20 22:08:54 GMT 2023 - 63.6K bytes - Viewed (6) -
tensorflow/c/experimental/filesystem/modular_filesystem_test.cc
return tensorflow::str_util::StripPrefix(absolute_path, root_dir_); } // Initializes the randomness used to ensure test isolation. static void InitializeTestRNG() { std::random_device rd; std::mt19937 gen(rd()); std::uniform_int_distribution<> distribution; rng_val_ = distribution(gen); } static void SetCloudPath(const std::string& cloud_path) { cloud_path_ = cloud_path;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 20:25:58 GMT 2022 - 71K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/ram_file_block_cache_test.cc
env->SetNowSeconds(now + 2); // The pruning thread periodically compares env->NowSeconds() with the oldest // block's timestamp to see if it should evict any files. At the current fake // timestamp of `now` + 2, file "a" is stale because its first block is stale, // but file "b" is not stale yet. Thus, once the pruning thread wakes up (in
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)