- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 10 for service (0.22 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.h
#ifndef TENSORFLOW_C_EAGER_PARALLEL_DEVICE_PARALLEL_DEVICE_H_ #define TENSORFLOW_C_EAGER_PARALLEL_DEVICE_PARALLEL_DEVICE_H_ #include "tensorflow/c/c_api.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h" namespace tensorflow { namespace parallel_device { // Allocate a parallel device named `device_name` which forwards operations to
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jun 04 21:49:16 GMT 2020 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device_testlib.h
TFE_Context* context, const char* device_name, const std::array<const char*, num_devices>& underlying_devices, TF_Status* status) { TFE_CustomDevice device; void* device_info; tensorflow::parallel_device::AllocateParallelDevice( device_name, underlying_devices.data(), underlying_devices.size(), &device, &device_info); TFE_RegisterCustomDevice(context, device, device_name, device_info, status); }
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 09 01:12:35 GMT 2021 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_distributed_manager.h
// Check if the remote task is alive. virtual Status CheckRemoteAlive(const std::string& remote_task_name, bool* is_alive) = 0; // Get pointer to the coordination service agent instance. virtual tsl::CoordinationServiceAgent* GetCoordinationServiceAgent() = 0; }; } // namespace tensorflow
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 2.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// `send_device`, `recv_device`, and `send_device_incarnation` set. TFE_Op* SendOp(TFE_Context* ctx, TFE_TensorHandle* in, const std::string& op_name, const std::string& send_device, const std::string& recv_device, tensorflow::uint64 send_device_incarnation); // Return a RecvOp op `op_name` with the attributes `send_device`,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Jul 17 23:43:59 GMT 2023 - 7.7K bytes - Viewed (0) -
tensorflow/c/eager/abstract_operation.h
// returned by DeviceName should be "/device:GPU:*" until a particular GPU is // chosen for the operation by the device placement logic in the // executor. After that, the value returned by DeviceName will be a full // device name such as "/job:localhost/replica:0/task:0/device:GPU:1". virtual const string& DeviceName() const = 0; // Sets the operation device name. //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
// C API for device. The API is under active development and eventually // should allow registering a plugin device with TensorFlow. #ifdef __cplusplus extern "C" { #endif // TF_Device is a C wrapper to the C++ TF Device class. This is to be passed // through TF_OpKernelContext, and is opaque to plugin. typedef struct TF_Device TF_Device; typedef struct TF_VariableInfo TF_VariableInfo;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Dec 20 20:01:06 GMT 2023 - 7.2K bytes - Viewed (0) -
tensorflow/c/eager/custom_device_testutil.h
limitations under the License. ==============================================================================*/ #ifndef TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_ #define TENSORFLOW_C_EAGER_CUSTOM_DEVICE_TESTUTIL_H_ // A simple logging device to test custom device registration. #include <memory> #include "tensorflow/c/c_api.h" #include "tensorflow/c/eager/c_api.h" #include "tensorflow/c/eager/c_api_experimental.h"
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 27 23:39:24 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/eager/immediate_execution_operation.h
virtual ImmediateExecutionContext* GetContext() const = 0; // Following two methods are used to support custom device. // Return true if the inputs contain custom device tensor handle. It means // that the argument need to be handled by a custom device. virtual bool HasCustomDeviceInput() const = 0; virtual const tensorflow::OpDef* OpDef() const = 0;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 22:40:32 GMT 2022 - 3.6K bytes - Viewed (0) -
tensorflow/c/c_test_util.h
TF_Operation* Mul(TF_Operation* l, TF_Operation* r, TF_Graph* graph, TF_Status* s, const char* name = "mul"); // If `op_device` is non-empty, set the created op on that device. TF_Operation* MinWithDevice(TF_Operation* l, TF_Operation* r, TF_Graph* graph, const string& op_device, TF_Status* s, const char* name = "min"); TF_Operation* Neg(TF_Operation* n, TF_Graph* graph, TF_Status* s,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Aug 09 01:06:53 GMT 2018 - 6K bytes - Viewed (0) -
tensorflow/c/eager/gradients_internal.h
// These APIs are mainly to facilitate testing and are subject to change. // Records the op name in the `ForwardOperation`. Status Reset(AbstractOperation*, const char* op, const char* raw_device_name, ForwardOperation*); // Records the inputs in the `ForwardOperation`. Status AddInput(AbstractOperation*, AbstractTensorHandle*, ForwardOperation*); Status AddInputList(AbstractOperation*,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0)