- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 13 for service (0.2 sec)
-
tensorflow/c/experimental/next_pluggable_device/c_api.cc
#include "tensorflow/core/tfrt/common/pjrt_util.h" #include "tsl/distributed_runtime/coordination/coordination_service_agent.h" TF_Device* TF_GetDevice(TF_OpKernelContext* ctx) { auto* cc_ctx = reinterpret_cast<tensorflow::OpKernelContext*>(ctx); return reinterpret_cast<TF_Device*>(cc_ctx->device()); } // -------------------------- Resource ---------------------------------------
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/dlpack.cc
DLDevice ctx; const char* device_name = tensorflow::unwrap(h)->BackingDeviceName(&status->status); DeviceNameUtils::ParsedName parsed_name; tensorflow::DeviceNameUtils::ParseFullName(device_name, &parsed_name); 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") {
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/experimental/grappler/grappler.cc
TF_RETURN_IF_ERROR(ValidateTPOptimizerConfigs(optimizer_configs)); CGraphOptimizerRegister( [=]() { return new CGraphOptimizer(optimizer, params.device_type); }, optimizer_configs, params.device_type); return absl::OkStatus(); } } // namespace grappler } // namespace tensorflow void TF_GetNodesToPreserveListSize(const TF_GrapplerItem* item, int* num_values,
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 06 19:12:29 GMT 2023 - 15K bytes - Viewed (1) -
tensorflow/c/eager/gradients.cc
namespace internal { Status Reset(AbstractOperation* op_, const char* op, const char* raw_device_name, ForwardOperation* forward_op_) { forward_op_->op_name = op; forward_op_->attrs.Reset(op); return op_->Reset(op, raw_device_name); } Status AddInput(AbstractOperation* op_, AbstractTensorHandle* input, ForwardOperation* forward_op_) {
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/eager/c_api_experimental.cc
#include "xla/tsl/c/tsl_status_internal.h" #include "xla/tsl/distributed_runtime/coordination/coordination_service_agent.h" #include "tensorflow/core/common_runtime/composite_device.h" #include "tensorflow/core/common_runtime/device.h" #include "tensorflow/core/common_runtime/eager/eager_operation.h" #include "tensorflow/core/distributed_runtime/coordination/coordination_service_error_util.h" #include "tensorflow/core/framework/function.h"
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/parallel_device/parallel_device.cc
const char* const* underlying_devices, int num_underlying_devices, TFE_CustomDevice* device, void** device_info) { device->copy_tensor_to_device = &CopyToParallelDevice; device->copy_tensor_from_device = &CopyTensorFromParallelDevice; device->delete_device = &DeleteParallelDevice; device->execute = &ParallelDeviceExecute;
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Mar 29 22:05:31 GMT 2023 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_lib.cc
absl::optional<int64_t> step_id) const { for (int device_index = 0; device_index < underlying_devices_.size(); ++device_index) { DeviceThread* device_thread = device_threads_[device_index].get(); std::vector<TFE_TensorHandle*> device_inputs; device_inputs.reserve(inputs.size()); for (int input_index = 0; input_index < inputs.size(); ++input_index) {
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/eager/c_api_distributed_test.cc
const string composite_device_name = "/job:localhost/replica:0/task:0/device:COMPOSITE:0"; EXPECT_EQ(TFE_TensorHandleDeviceName(packed_handle, status), composite_device_name); EXPECT_EQ(TFE_TensorHandleBackingDeviceName(packed_handle, status), composite_device_name); ASSERT_EQ(TF_GetCode(status), TF_OK) << TF_Message(status);
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/experimental/grappler/grappler_test.cc
TF_SetStatus(status, TF_OK, ""); PopulateDefaultParam(params); params->device_type = "Device0"; }; auto plugin_init_1 = [](TP_OptimizerRegistrationParams* const params, TF_Status* const status) -> void { TF_SetStatus(status, TF_OK, ""); PopulateDefaultParam(params); params->device_type = "Device1"; }; TF_ASSERT_OK(InitGraphPlugin(plugin_init_0));
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Thu Apr 13 22:30:58 GMT 2023 - 11.6K bytes - Viewed (0) -
tensorflow/c/eager/c_api.cc
const char* device_name, void* device_info, TF_Status* status) { // Fill in default values for optional functionality. if (device.pack == nullptr) { device.pack = &DefaultCustomDevicePack; } auto custom_device = std::make_unique<tensorflow::CustomDeviceAPI>( ctx, device, device_info, device_name);
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)