- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for Enum (0.15 sec)
-
tensorflow/c/eager/abstract_function.h
// representation between various supported formats e.g. FunctionDef and Mlir // function. class AbstractFunction : public core::RefCounted { protected: enum AbstractFunctionKind { kGraph, kMlir }; explicit AbstractFunction(AbstractFunctionKind kind) : kind_(kind) {} public: // Returns which subclass is this instance of.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Mar 04 19:49:06 GMT 2024 - 1.9K bytes - Viewed (0) -
tensorflow/c/eager/c_api_experimental.h
// Custom devices must use TFE_NewCustomDeviceTensorHandle instead. TF_CAPI_EXPORT extern TFE_TensorHandle* TFE_NewTensorHandleFromDeviceMemory( TFE_Context* ctx, const char* device_name, TF_DataType, const int64_t* dims, int num_dims, void* data, size_t len, void (*deallocator)(void* data, size_t len, void* arg), void* deallocator_arg, TF_Status* status); // Retrieves the address space (i.e. job, replia, task) of the local host and
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_context.h
namespace tensorflow { class EagerExecutor; class EagerContext; class CustomDevice; class CustomDeviceOpHandler; class Device; // LINT.IfChange // Note: Keep in sync with exported copy of enum in eager/c_api.h. enum ContextDevicePlacementPolicy { // Running operations with input tensors on the wrong device will fail. DEVICE_PLACEMENT_EXPLICIT = 0, // Copy the tensor to the right device but log a warning.
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Jul 06 08:34:00 GMT 2023 - 12.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api_test_util.h
// Create a ServerDef with the given `job_name` and add `num_tasks` tasks in it. tensorflow::ServerDef GetServerDef(const tensorflow::string& job_name, int num_tasks); // Create a ServerDef with job name "localhost" and add `num_tasks` tasks in it. tensorflow::ServerDef GetServerDef(int num_tasks); // Create a multi-client ServerDef with the given `job_name`, add `num_tasks`
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
const int64_t* dims, int num_dims); // `dims` and `num_dims` must point to arrays of length `num_shapes`. // Set `num_dims[i]` to -1 to represent "unknown rank". Otherwise, // `dims[i]` points to an array of length `num_dims[i]`. `dims[i][j]` // must be >= -1, with -1 meaning "unknown dimension". TF_CAPI_EXPORT extern void TF_SetAttrShapeList(TF_OperationDescription* desc,
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/experimental/next_pluggable_device/c_api.h
// 0 for num_options if no options need to be set. You can use // ConvertToPjRtNamedValueList in // tensorflow/compiler/xla/pjrt/c/pjrt_c_api_helpers.h to generate the options. TF_CAPI_EXPORT extern void TF_CreateAndSetPjRtCApiClient( const char* device_type, TF_Status* status, PJRT_NamedValue* create_options, int num_options);
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/unified_api_testutil.h
// Return a tensor handle with given type, values and dimensions. template <class T, TF_DataType datatype> Status TestTensorHandleWithDims(AbstractContext* ctx, const T* data, const int64_t* dims, int num_dims, AbstractTensorHandle** tensor) { std::unique_ptr<TF_Status, decltype(&TF_DeleteStatus)> status( TF_NewStatus(), TF_DeleteStatus); TFE_Context* eager_ctx =
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0)