- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for sesd (0.14 sec)
-
tensorflow/c/c_api_experimental.cc
// CPU tensor to GPU). // Setting a larger thread pool does not help with the Swift caller, as we use // a different TFE context for each thread of execution (for running graph // functions, and their send/recvs corountines). config.set_inter_op_parallelism_threads(1); TF_Buffer* ret = TF_NewBuffer(); TF_CHECK_OK(MessageToBuffer(config, ret)); return ret; }
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/eager/c_api_test_util.cc
TFE_OpSetAttrString(op, "tensor_name", "dummy", 5); TFE_OpSetAttrString(op, "send_device", send_device.c_str(), send_device.size()); TFE_OpSetAttrString(op, "recv_device", recv_device.c_str(), recv_device.size()); TFE_OpSetAttrInt(op, "send_device_incarnation", send_device_incarnation); return op; }
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 23.5K bytes - Viewed (2) -
tensorflow/c/eager/c_api_experimental.cc
if (prefix == nullptr || strlen(prefix) == 0) prefix = "/job:localhost/replica:0/task:0"; tensorflow::SessionOptions sess_options; (*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.
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/eager/c_api_remote_test_util.cc
TFE_TensorHandle* h0_task0 = TestMatrixTensorHandle(ctx); TFE_TensorHandle* h1_task0 = TestMatrixTensorHandle(ctx); std::vector<TFE_TensorHandle*> handles_task0; if (heavy_load_on_streaming_rpc) { // Send 50 tensor copy requests to simulate that there have been some RPC // requests been enqueued. for (int i = 0; i < 50; ++i) { handles_task0.push_back(TestMatrixTensorHandle(ctx)); } }
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/c_api_test.cc
TF_DeleteGraph); TF_SessionOptions* opts = TF_NewSessionOptions(); TF_Session* sess = TF_NewSession(graph.get(), opts, s); TF_DeleteSessionOptions(opts); const string gpu_device_name = GPUDeviceName(sess); TF_DeleteSession(sess, s); CHECK_EQ(TF_OK, TF_GetCode(s)) << TF_Message(s); return gpu_device_name; }
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)