- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for shallow (0.16 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/c_api_experimental.h
// `enable_xla_compilation` is non-zero, and OFF otherwise. // b) ConfigProto.gpu_options.allow_growth is set to `gpu_memory_allow_growth`. // c) ConfigProto.device_count is set to `num_cpu_devices`. TF_CAPI_EXPORT extern TF_Buffer* TF_CreateConfig( unsigned char enable_xla_compilation, unsigned char gpu_memory_allow_growth, unsigned int num_cpu_devices);
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/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)