- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 72 for device (0.21 sec)
-
tensorflow/c/eager/parallel_device/parallel_device.h
void AllocateParallelDevice(const char* device_name, const char* const* underlying_devices, int num_underlying_devices, TFE_CustomDevice* device, void** device_info); } // namespace parallel_device } // namespace tensorflow
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/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/parallel_device/parallel_device_testlib.h
} template <std::size_t num_devices> void RegisterParallelDevice( 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);
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
std::unique_ptr<ParallelTensor> DeviceIDs(TFE_Context* context, TF_Status* status) const; // The number of devices operations run on. size_t num_underlying_devices() const { return underlying_devices_.size(); } // The devices operations run on. const std::vector<std::string>& underlying_devices() const { return underlying_devices_;
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_experimental.h
// // `device_name` must not name an existing physical or custom device. It must // follow the format: // // /job:<name>/replica:<replica>/task:<task>/device:<type>:<device_num> // // If the device is successfully registered, `status` is set to TF_OK. Otherwise // the device is not usable. In case of a bad status, `device.delete_device` is // still called on `device_info` (i.e. the caller does not retain ownership).
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Wed Feb 21 22:37:46 GMT 2024 - 39.5K 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/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/experimental/grappler/grappler.h
// void TF_InitGraph(TP_OptimizerRegistrationParams* params, // TF_Status* status) { // params->struct_size = TP_OPTIMIZER_REGISTRATION_PARAMS_STRUCT_SIZE; // params->device_type = "MY_DEVICE"; // // // Disable certain optimizer. // params->optimizer_configs->struct_size = // TP_OPTIMIZER_CONFIGS_STRUCT_SIZE; params->optimizer_configs->remapping = // TF_TriState_Off; //
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Aug 03 18:08:43 GMT 2022 - 12.5K 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/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)