- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for OpTapeEntry (0.08 sec)
-
tensorflow/c/eager/gradients.h
// `GradientTape::Watch`.) The op_id is simply a unique index assigned to each // op executed under the tape. A separate map (`tensorflow::eager::OpTape`) // maintains the map from `op_id` to a `OpTapeEntry` which stores the `op_type`, // inputs and outputs and the gradient function These data structures combined // allow us to trace the data dependencies between operations and hence compute // gradients. //
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Sep 26 10:27:05 UTC 2022 - 6.9K bytes - Viewed (0) -
tensorflow/c/eager/tape.h
#include "tensorflow/core/platform/types.h" namespace tensorflow { namespace eager { // Represents an entry in the tape. template <typename BackwardFunction, typename TapeTensor> struct OpTapeEntry { string op_type; std::vector<TapeTensor> output_tensor_info; std::vector<int64_t> input_tensor_id; // TODO(apassos) consider narrowing down this interface. BackwardFunction* backward_function;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 02 12:40:29 UTC 2024 - 47.2K bytes - Viewed (0)