- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for clinit (0.14 sec)
-
tensorflow/c/c_api.cc
RunMetadata run_metadata_proto; result = session->Run(run_options_proto, input_pairs, output_tensor_names, target_oper_names, &outputs, &run_metadata_proto); // Serialize back to upstream client, who now owns the new buffer if (run_metadata != nullptr) { status->status = MessageToBuffer(run_metadata_proto, run_metadata); if (!status->status.ok()) return; } } else {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Apr 15 03:35:10 GMT 2024 - 102.3K bytes - Viewed (0) -
tensorflow/c/eager/gradients.cc
return errors::InvalidArgument("Value specified for `", attr_name, "` has ", num_dims, " dimensions which is over the limit of ", TensorShape::MaxDimensions(), "."); } TensorShapeProto proto; if (num_dims < 0) { proto.set_unknown_rank(true); } else {
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/tensor_pjrt_buffer_util_test.cc
CHECK_OK(pjrt::SetPjrtApi(DEVICE_CPU, GetPjrtApi())); } auto pjrt_client = xla::GetCApiClient(DEVICE_CPU); CHECK_OK(pjrt_client.status()); auto c_api_client = down_cast<xla::PjRtCApiClient*>(pjrt_client->get()); std::vector<int32_t> data(1, 0); xla::Shape shape = xla::ShapeUtil::MakeShape(xla::S32, {1}); auto buffer = c_api_client->pjrt_c_client()->client->BufferFromHostBuffer(
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/experimental/next_pluggable_device/c_api.cc
if (!pjrt_c_api_client.ok()) { status->status = pjrt_c_api_client.status(); return nullptr; } status->status = absl::OkStatus(); return (*pjrt_c_api_client)->pjrt_c_client(); } PJRT_Buffer* TF_GetPjRtCBuffer(TF_Tensor* c_tensor, TF_Status* status) { tensorflow::Tensor tensor; auto s = tensorflow::TF_TensorToTensor(c_tensor, &tensor); if (!s.ok()) { status->status = s;
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jan 09 00:52:04 GMT 2024 - 13.9K bytes - Viewed (1) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
// parallel device. device_(device), executor_( TFE_NewExecutor(is_async, /*enable_streaming_enqueue=*/true, /*in_flight_nodes_limit=*/in_flight_nodes_limit)), op_(nullptr), thread_(tensorflow::Env::Default()->StartThread( tensorflow::ThreadOptions(), "parallel_device_execute", std::bind(&DeviceThread::Run, this))) {}
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util.cc
absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type) { TF_ASSIGN_OR_RETURN(tsl::StatusOr<xla::PjRtClient*> pjrt_client, tensorflow::GetPjRtClient(device_type)); auto* pjrt_c_api_client = dynamic_cast<xla::PjRtCApiClient*>(*pjrt_client); if (pjrt_c_api_client == nullptr) { return absl::InternalError(absl::StrCat("PjRtClient for ", device_type.type_string(),
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 30 19:20:20 GMT 2023 - 3.7K bytes - Viewed (0) -
tensorflow/c/experimental/filesystem/plugins/gcs/gcs_filesystem.cc
TF_SetStatus(status, TF_OK, ""); } void Init(TF_Filesystem* filesystem, TF_Status* status) { google::cloud::StatusOr<gcs::Client> client = gcs::Client::CreateDefaultClient(); if (!client) { TF_SetStatusFromGCSStatus(client.status(), status); return; } filesystem->plugin_filesystem = new GCSFile(std::move(client.value())); TF_SetStatus(status, TF_OK, ""); }
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) -
tensorflow/c/eager/c_api_experimental.cc
status->status = absl::OkStatus(); } TFE_Executor* TFE_NewExecutor(bool is_async, bool enable_streaming_enqueue, int in_flight_nodes_limit) { return new TFE_Executor(is_async, enable_streaming_enqueue, in_flight_nodes_limit); } void TFE_DeleteExecutor(TFE_Executor* executor) { delete executor; } bool TFE_ExecutorIsAsync(TFE_Executor* executor) {
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_test_util.cc
auto virtual_devices = gpu_options->mutable_experimental()->add_virtual_devices(); for (int i = 0; i < num_virtual_gpus; ++i) { virtual_devices->add_memory_limit_mb(200); } } return server_def; } TFE_TensorHandle* CreateVarHandle(TFE_Context* ctx, const tensorflow::string& device_name,
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.cc
ctx, keep_alive_secs, proto, proto_len, /*init_timeout_in_ms=*/0, /*retries=*/0, status, /*clear_existing_contexts=*/false); } // Set server def with timeout. TF_CAPI_EXPORT extern void TFE_ContextSetServerDefWithTimeout( TFE_Context* ctx, int keep_alive_secs, const void* proto, size_t proto_len, int64_t init_timeout_in_ms, TF_Status* status, bool clear_existing_contexts) {
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)