- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for axis (0.13 sec)
-
tensorflow/c/eager/c_api_test_util.h
// Return a 1-D INT32 tensor containing a single value 1. TFE_TensorHandle* TestAxisTensorHandle(TFE_Context* ctx); // Return an op taking minimum of `input` long `axis` dimension. TFE_Op* MinOp(TFE_Context* ctx, TFE_TensorHandle* input, 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".
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/c_api_experimental.h
// this context. TF_CAPI_EXPORT extern void TFE_ContextDisableGraphCollection(TFE_Context* ctx); // TODO(fishx): Move these monitoring APIs into a separate file. // ----------------------------------------------------------------------------- // Monitoring Counter APIs. // These APIs de-templated monitoring Counter for swig. typedef struct TFE_MonitoringCounterCell TFE_MonitoringCounterCell;
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_tensor_handle.h
// Returns number of elements across all dimensions. virtual Status NumElements(int64_t* num_elements) const = 0; // Returns size of specified dimension // // -1 indicates an unknown axis length; this is unreachable for most standard // ImmediateExecutionTensorHandles, but comes up for example when computing // the shape of a parallel tensor with component shapes differing across // devices.
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) -
tensorflow/c/eager/c_api_unified_experimental.h
extern "C" { #endif // ============================================================================= // Unified Execution APIs for Eager and tracing backends. // ============================================================================= // ----------------------------------------------------------------------------- // Core APIs // -----------------------------------------------------------------------------
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:00 GMT 2021 - 7K bytes - Viewed (0) -
tensorflow/c/c_api.h
// // On success, place OK in status. // On failure, place an error status in status. TF_CAPI_EXPORT extern void TF_RegisterFilesystemPlugin( const char* plugin_filename, TF_Status* status); // Apis that are corresponding to python c api. -------------------- // Add control input to `op`. TF_CAPI_EXPORT extern void TF_AddOperationControlInput(TF_Graph* graph,
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/gradients_internal.h
namespace tensorflow { namespace gradients { namespace internal { // Helper functions which delegate to `AbstractOperation`, update // the state of the ForwardOperation and call the tape as appropriate. // 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*);
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) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
// --------------------- Coordination service -------------------------------- // Returns a not owning pointer to the coordination service agent, which is // opaque to plugin. Plugin OpKernels need to use the accompanying C APIs to // access coordination service functionalities. TF_CAPI_EXPORT extern TF_CoordinationServiceAgent* TF_GetCoordinationServiceAgent(TF_OpKernelContext* ctx);
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/immediate_execution_context.h
// TF eager runtime. // TODO(b/172877902): These helper functions are currently used to support // PyFuncOp on TFRT, and might be useful for ops that directly use low // level TF APIs. Remove/replace the following functions when TFRT native // ops are implemented. //===--------------------------------------------------------------------===// // Create an abstract tensor handle from tensorflow::Tensor.
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)