- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 39 for Service (0.19 sec)
-
tensorflow/c/eager/parallel_device/parallel_device_remote_test.cc
const char* first_device = "/job:worker/replica:0/task:1/device:CPU:0"; const char* second_device = "/job:worker/replica:0/task:2/device:CPU:0"; const char* device_name = "/job:localhost/replica:0/task:0/device:CUSTOM:0"; std::array<const char*, 2> underlying_devices{first_device, second_device}; RegisterParallelDevice(context.get(), device_name, underlying_devices, status.get());
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 22:09:57 GMT 2023 - 6.7K bytes - Viewed (0) -
tensorflow/c/eager/dlpack_test.cc
ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); const DLTensor* dltensor_out = &dlm_out->dl_tensor; EXPECT_EQ(dltensor_out->device.device_type, dltensor_in->device.device_type); EXPECT_EQ(dltensor_out->device.device_id, dltensor_in->device.device_id); EXPECT_EQ(dltensor_out->ndim, dltensor_in->ndim); EXPECT_EQ(dltensor_out->dtype.code, dltensor_in->dtype.code);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Jun 30 03:04:46 GMT 2023 - 4.4K bytes - Viewed (0) -
tensorflow/c/c_api.cc
} void TF_SetRequestedDevice(TF_Graph* graph, TF_Operation* op, const char* device) { using tensorflow::RecordMutation; mutex_lock l(graph->mu); op->node.set_requested_device(device); RecordMutation(graph, *op, "setting device"); } void TF_RemoveAllControlInputs(TF_Graph* graph, TF_Operation* op) { mutex_lock l(graph->mu);
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/c_api_unified_experimental.cc
void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type, TF_Status* s) { tsl::Set_TF_Status_from_Status( s, unwrap(op)->Reset(op_type, /*raw_device_name=*/nullptr)); } void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name, TF_Status* s) { TracingOperation* tracing_op = dyn_cast<TracingOperation>(unwrap(op));
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 9K bytes - Viewed (0) -
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/parallel_device/parallel_device_lib_test.cc
#include "tensorflow/c/eager/parallel_device/parallel_device_lib.h" #include "tensorflow/c/c_api.h" #include "tensorflow/c/c_api_experimental.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" #include "tensorflow/c/eager/parallel_device/parallel_device_testlib.h" #include "tensorflow/c/eager/tfe_context_internal.h"
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 08 23:47:35 GMT 2021 - 15.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_cluster_test.cc
// Check that copying it to the old remote device (named localhost) fails. TFE_TensorHandleCopyToDevice(h0_task0_new, ctx, remote_device_name, status); EXPECT_NE(TF_OK, TF_GetCode(status)) << TF_Message(status); // Copying and executing on the new remote device works. const char new_remote_device_name[] = "/job:worker/replica:0/task:1/device:CPU:0"; const char new_local_device_name[] =
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Apr 14 10:03:59 GMT 2023 - 19.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_remote_test.cc
const char remote_device_name[] = "/job:localhost/replica:0/task:1/device:CPU:0"; auto* h0_task1 = TFE_TensorHandleCopyToDevice(h0_task0, ctx, remote_device_name, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status); auto* h1_task1 = TFE_TensorHandleCopyToDevice(h1_task0, ctx, remote_device_name, status); ASSERT_EQ(TF_OK, TF_GetCode(status)) << TF_Message(status);
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Aug 12 00:14:22 GMT 2020 - 5.4K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.cc
#include "tensorflow/core/common_runtime/eager/context.h" #include "tensorflow/core/common_runtime/eager/eager_operation.h" #include "tensorflow/core/common_runtime/pluggable_device/pluggable_device_plugin_init.h" #include "tensorflow/core/distributed_runtime/rpc/grpc_server_lib.h" #include "tensorflow/core/framework/collective.h" #include "tensorflow/core/framework/node_def.pb.h"
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/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)