- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for cpu (0.13 sec)
-
tensorflow/c/c_api_experimental.cc
flags->tf_xla_cpu_global_jit = true; flags->tf_xla_min_cluster_size = 1; } else { optimizer_options->set_global_jit_level(tensorflow::OptimizerOptions::OFF); } auto* gpu_options = config.mutable_gpu_options(); gpu_options->set_allow_growth(gpu_memory_allow_growth); (*config.mutable_device_count())["CPU"] = num_cpu_devices;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 29.4K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
StatusIs(error::INTERNAL, HasSubstr(absl::StrCat("PjRtClient for ", DEVICE_CPU, " is not type PjRtCApiClient")))); } TEST(TensorPjRtBufferUtilTest, GetPjRtCApiClientSuccess) { auto status = pjrt::PjrtApi(DEVICE_CPU); if (!status.ok()) { TF_ASSERT_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/dlpack.cc
std::string device_type = parsed_name.type; int device_id = 0; if (parsed_name.has_id) { device_id = parsed_name.id; } ctx.device_id = device_id; if (device_type == "CPU") { ctx.device_type = DLDeviceType::kDLCPU; } else if (device_type == "GPU") { #if TENSORFLOW_USE_ROCM ctx.device_type = DLDeviceType::kDLROCM; #else ctx.device_type = DLDeviceType::kDLCUDA; #endif
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 12.8K bytes - Viewed (0) -
tensorflow/c/eager/c_api_distributed_test.cc
EXPECT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status); const char task0_name[] = "/job:localhost/replica:0/task:0/device:CPU:0"; const char task1_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; const char task2_name[] = "/job:localhost/replica:0/task:2/device:CPU:0"; // Create one variable per task. TFE_TensorHandle* h0 = TestVariable(ctx, 1.0, task1_name);
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_experimental_test.cc
ASSERT_TRUE(absl::StrContains(device_type_cpu, "CPU")) << device_type_cpu; int device_id_cpu = TFE_TensorHandleDeviceID(h_cpu, status.get()); ASSERT_EQ(TF_OK, TF_GetCode(status.get())) << TF_Message(status.get()); ASSERT_EQ(0, device_id_cpu) << device_id_cpu; TFE_DeleteTensorHandle(h_default); TFE_DeleteTensorHandle(h_cpu); TFE_Executor* executor = TFE_ContextGetExecutorForThread(ctx);
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/eager/c_api_test.cc
TFE_DeleteTensorHandle(retvals[0]); TFE_DeleteContext(ctx); TF_DeleteStatus(status); } TEST(CAPI, Execute_MatMul_CPU_Runtime_Error) { Execute_MatMul_CPU_Runtime_Error(false); } TEST(CAPI, Execute_MatMul_CPU_Runtime_ErrorAsync) { Execute_MatMul_CPU_Runtime_Error(true); } void Execute_MatMul_CPU_Type_Error(bool async) { TF_Status* status = TF_NewStatus(); TFE_ContextOptions* opts = TFE_NewContextOptions();
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.cc
(*sess_options.config.mutable_device_count())["CPU"] = num_cpus; status->status = tensorflow::DeviceFactory::AddCpuDevices(sess_options, prefix, &devices); // Remove the device that has the host device name since host device is alreay // in an initialized context. for (auto d = devices.begin(); d != devices.end();) { if (absl::StrContains(d->get()->name(), "CPU:0")) { d = devices.erase(d); } else {
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/c_api_test.cc
TF_DeleteStatus(s); } void Deallocator(void* data, size_t, void* arg) { tensorflow::cpu_allocator()->DeallocateRaw(data); *reinterpret_cast<bool*>(arg) = true; } TEST(CAPI, Tensor) { const int num_bytes = 6 * sizeof(float); float* values = reinterpret_cast<float*>(tensorflow::cpu_allocator()->AllocateRaw( TF_TensorDefaultAlignment(), num_bytes)); int64_t dims[] = {2, 3};
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_function_test.cc
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_); ASSERT_NE(*op, nullptr); } FunctionDef fdef() {
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/eager/c_api.cc
#endif // !IS_MOBILE_PLATFORM using tensorflow::string; namespace { string DeviceName(const tensorflow::Device* d) { return (d == nullptr) ? "cpu:0" : d->name(); } // Annotate eager runtime construction context to the given `function_def` as // an attribute. void AnnotateEagerRuntimeConstructionContext( tensorflow::FunctionDef& function_def) {
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)