- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for zeros (0.16 sec)
-
tensorflow/c/eager/tape.h
return absl::OkStatus(); } // We may need to allocate zero inputs for trainable dtypes we don't have JVPs // for. Make sure they get cleaned up. std::vector<Gradient*> new_zeros; auto delete_new_zeros = gtl::MakeCleanup([&new_zeros, this] { for (Gradient* tensor : new_zeros) { this->vspace_.DeleteGradient(tensor); } }); std::vector<Gradient*> in_grads;
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Apr 02 12:40:29 GMT 2024 - 47.2K bytes - Viewed (1) -
tensorflow/c/eager/gradients.h
// allow us to trace the data dependencies between operations and hence compute // gradients. // // `ZerosLike` is not expected to be called and returns a nullptr. The creation // of default zeros grads is handled by the `DefaultGradientFunction` registered // for each op. // TODO(srbs): We need to define `ZerosLike` here to keep the compiler happy. // Figure out a way to avoid this.
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/env.h
typedef struct TF_Thread TF_Thread; typedef struct TF_ThreadOptions { // Thread stack size to use (in bytes), zero implies that the system default // will be used. size_t stack_size; // Guard area size to use near thread stacks to use (in bytes), zero implies // that the system default will be used. size_t guard_size;
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Sat Jan 09 02:53:27 GMT 2021 - 9.6K bytes - Viewed (0) -
tensorflow/c/c_api.h
// // `num_return_outputs` must be the number of return outputs added (i.e. the // result of TF_ImportGraphDefOptionsNumReturnOutputs()). If // `num_return_outputs` is non-zero, `return_outputs` must be of length // `num_return_outputs`. Otherwise it can be null. TF_CAPI_EXPORT extern void TF_GraphImportGraphDefWithReturnOutputs( TF_Graph* graph, const TF_Buffer* graph_def,
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Oct 26 21:08:15 GMT 2023 - 82.3K bytes - Viewed (3) -
tensorflow/c/eager/unified_api_testutil.h
absl::Span<AbstractTensorHandle* const> inputs, std::vector<AbstractTensorHandle*>* params); // A callable that takes tensor inputs and returns zero or more tensor outputs. using Model = std::function<Status(AbstractContext*, absl::Span<AbstractTensorHandle* const>, absl::Span<AbstractTensorHandle*>)>;
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 4K bytes - Viewed (0) -
tensorflow/c/c_api_experimental.h
unsigned char should_enable); // Create a serialized tensorflow.ConfigProto proto, where: // // a) ConfigProto.optimizer_options.global_jit_level is set to ON_1 if // `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(
C - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Thu Apr 27 21:07:00 GMT 2023 - 15.1K bytes - Viewed (0)