- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for devices (0.23 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/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/c_api.h
TF_ImportGraphDefOptions* opts, const char* prefix); // Set the execution device for nodes in `graph_def`. // Only applies to nodes where a device was not already explicitly specified. // `device` is copied and has no lifetime requirements. TF_CAPI_EXPORT extern void TF_ImportGraphDefOptionsSetDefaultDevice( TF_ImportGraphDefOptions* opts, const char* device); // Set whether to uniquify imported operation names. If true, imported operation
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
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/BUILD
) # Config setting that is satisfied when CUDA device code should be compiled # with clang. It does not imply that CUDA support has been enabled. alias( name = "is_cuda_compiler_clang", actual = if_oss( "@local_config_cuda//:is_cuda_compiler_clang", "@local_config_cuda//cuda:TRUE", ), ) # Config setting that is satisfied when CUDA device code should be compiled
Plain Text - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 09 18:15:11 GMT 2024 - 53.4K bytes - Viewed (8) -
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/filesystem_interface.h
/// /// Since memory that is allocated by the DSO gets transferred to core /// TensorFlow, we need to provide a way for the allocation and deallocation to /// match. This is why this structure also defines `plugin_memory_allocate` and /// `plugin_memory_free` members. /// /// All memory allocated by the plugin that will be owned by core TensorFlow /// must be allocated using the allocator in this structure. Core TensorFlow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri May 27 17:36:54 GMT 2022 - 53.1K bytes - Viewed (0) -
configure.py
Returns: string of native cuda compute capabilities, separated by comma. """ device_query_bin = os.path.join( environ_cp.get('CUDA_TOOLKIT_PATH'), 'extras/demo_suite/deviceQuery') if os.path.isfile(device_query_bin) and os.access(device_query_bin, os.X_OK): try: output = run_shell(device_query_bin).split('\n') pattern = re.compile('[0-9]*\\.[0-9]*')
Python - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 18:25:36 GMT 2024 - 53.8K bytes - Viewed (1) -
.bazelrc
build:windows --copt=/Zc:__cplusplus build:windows --host_copt=/Zc:__cplusplus # Tensorflow uses M_* math constants that only get defined by MSVC headers if # _USE_MATH_DEFINES is defined. build:windows --copt=/D_USE_MATH_DEFINES build:windows --host_copt=/D_USE_MATH_DEFINES # Windows has a relatively short command line limit, which TF has begun to hit. # See https://docs.bazel.build/versions/main/windows.html
Plain Text - Registered: Tue May 07 12:40:20 GMT 2024 - Last Modified: Thu May 02 19:34:20 GMT 2024 - 52.8K bytes - Viewed (2)