- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for device_type (0.2 sec)
-
tensorflow/c/eager/c_api_test_util.h
TFE_TensorHandle* axis); // If there is a device of type `device_type`, returns true // and sets 'device_name' accordingly. // `device_type` must be either "GPU" or "TPU". bool GetDeviceName(TFE_Context* ctx, tensorflow::string* device_name, const char* device_type); // Create a ServerDef with the given `job_name` and add `num_tasks` tasks in it.
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/experimental/next_pluggable_device/c_api.h
TF_CAPI_EXPORT extern void TF_CreateAndSetPjRtCApiClient( const char* device_type, TF_Status* status, PJRT_NamedValue* create_options, int num_options); // Resets the PjRt client for a device. After this, `TF_GetPjRtCClient` will // returns an error for that device. TF_CAPI_EXPORT extern void TF_ResetPjRtCClient(const char* device_type, TF_Status* status);
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/experimental/next_pluggable_device/tensor_pjrt_buffer_util.h
Tensor* tensor); absl::StatusOr<xla::PjRtCApiClient*> GetPjRtCApiClient( const DeviceType& device_type); absl::Status ResetPjRtClient(const DeviceType& device_type); } // namespace tensorflow
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Mon Oct 09 19:12:59 GMT 2023 - 1.5K 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 // needs to be updated with the argument length of the input list.
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/grappler/grappler_internal.h
using OwnedTFBuffer = std::unique_ptr<TF_Buffer, TFBufferDeleter>; class CGraphOptimizer : public CustomGraphOptimizer { public: explicit CGraphOptimizer(TP_Optimizer optimizer, const char* device_type) : optimizer_(optimizer), device_type_(device_type) { if (optimizer.create_func != nullptr) { c_optimizer_ = (*optimizer_.create_func)(); } else { c_optimizer_ = nullptr; } }
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Jun 08 08:58:23 GMT 2022 - 3.5K bytes - Viewed (1) -
tensorflow/c/eager/immediate_execution_tensor_handle.h
// Returns the device where the tensor was placed. virtual const char* BackingDeviceName(Status* status) const = 0; // Returns the device type which created the handle. virtual const char* DeviceType(Status* status) const = 0; // Returns the device ID which created the handle. virtual int DeviceId(Status* status) const = 0; // Returns a tensor for the handle. If tensor is remote, it will be copied.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Fri Mar 10 21:56:24 GMT 2023 - 4.3K bytes - Viewed (0)