- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for shallow (0.17 sec)
-
tensorflow/c/eager/c_api_experimental.h
// to pin to the physical device. // // This function is guaranteed to be called only when all of the custom-device // inputs are on this device. bool (*shall_pin_to_this_device)(const TFE_Op* op, TF_Status* s) = nullptr; } TFE_CustomDevice; // Registers a custom device for use with eager execution. // // Eager operations may be placed on this device, e.g. `with
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/tape.h
// // Gradient is the type returned by gradient functions. In Python TF it's either // Tensor or IndexedSlices or None, which here we map to nullptr. Gradients need // to allow their size to be computed and they need to be passable to a backward // function and deleted (as the backprop code creates lots of gradients the user // is not interested in). //
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/experimental/next_pluggable_device/c_api.h
#include "tensorflow/c/tf_status.h" #include "xla/pjrt/c/pjrt_c_api.h" // -------------------------------------------------------------------------- // C API for device. The API is under active development and eventually // should allow registering a plugin device with TensorFlow. #ifdef __cplusplus extern "C" { #endif // TF_Device is a C wrapper to the C++ TF Device class. This is to be passed // through TF_OpKernelContext, and is opaque to plugin.
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)