- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 22 for advice (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/parallel_device/parallel_device_lib.h
for (int device_index = 0; device_index < num_underlying_devices(); ++device_index) { auto device_value = absl::make_unique<DataType>(); *device_value = values[device_index]; std::unique_ptr<TF_Tensor, decltype(&TF_DeleteTensor)> tensor( TF_NewTensor( datatype_enum, /*dims=*/nullptr, /*num_dims=*/0, device_value.release(), sizeof(DataType),
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 25 15:21:13 GMT 2023 - 12.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/c_api.h
TF_ImportGraphDefOptions* opts, const char* prefix); // Set the execution device for nodes in `graph_def`. // Only applies to nodes where a device was not already explicitly specified. // `device` is copied and has no lifetime requirements. TF_CAPI_EXPORT extern void TF_ImportGraphDefOptionsSetDefaultDevice( TF_ImportGraphDefOptions* opts, const char* device); // Set whether to uniquify imported operation names. If true, imported operation
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
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/eager/c_api.h
// Controls how to act when we try to run an operation on a given device but // some input tensors are not on that device. // LINT.IfChange // Note: Keep in sync with internal copy of enum in eager/context.h. typedef enum TFE_ContextDevicePlacementPolicy { // Running operations with input tensors on the wrong device will fail. TFE_DEVICE_PLACEMENT_EXPLICIT = 0, // Copy the tensor to the right device but log a warning. TFE_DEVICE_PLACEMENT_WARN = 1,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 22.8K bytes - Viewed (1) -
tensorflow/c/eager/c_api_internal.h
// c_api_experimental.h struct TFE_ContextOptions { TF_SessionOptions session_options; // true if async execution is enabled. bool async = false; TFE_ContextDevicePlacementPolicy device_placement_policy{ TFE_DEVICE_PLACEMENT_SILENT}; // If true, use TFRT backend bool use_tfrt = false; // Whether to run elementary eager ops wrapped in a call op. bool run_eager_op_as_function = false;
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Jan 18 19:26:34 GMT 2023 - 1.9K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
int num_values); // Checks the tensorflow::NodeDef built via the methods above to see if it can // run on device_type. TF_CAPI_EXPORT extern void TF_AttrBuilderCheckCanRunOnDevice( TF_AttrBuilder* builder, const char* device_type, TF_Status* status); // For argument number input_index, fetch the corresponding number_attr that
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
namespace gradients { class TapeOperation : public AbstractOperation { public: explicit TapeOperation(AbstractOperation*, Tape*, const GradientRegistry&); void Release() override; Status Reset(const char* op, const char* raw_device_name) override; const string& Name() const override; const string& DeviceName() const override; Status SetDeviceName(const char* name) override; Status AddInput(AbstractTensorHandle* input) override;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1)