- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for RecordOperation (0.27 sec)
-
tensorflow/c/eager/tape.h
// // op_type is used to decide which of the incoming gradients can be left as // nullptr instead of building zeros when build_default_zeros_grads == true. void RecordOperation( const string& op_type, const std::vector<TapeTensor>& output_tensors, absl::Span<const int64_t> input_tensor_id, absl::Span<const tensorflow::DataType> input_dtypes,
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/eager/gradients_internal.h
Status SetAttrFunctionList(AbstractOperation*, const char* attr_name, absl::Span<const AbstractOperation*> values, ForwardOperation*); // Make the call to `Tape::RecordOperation`. Status Execute(AbstractOperation*, AbstractContext*, absl::Span<AbstractTensorHandle*> retvals, int* num_retvals, ForwardOperation*, Tape*, const GradientRegistry&);
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/eager/gradients.h
// on the tape and marks all its outputs as watched if at // least one input of the op is watched and has a trainable dtype. // op_name is optional and is used for debugging only. void RecordOperation(absl::Span<AbstractTensorHandle* const> inputs, absl::Span<AbstractTensorHandle* const> outputs, GradientFunction* gradient_function,
C - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Mon Sep 26 10:27:05 GMT 2022 - 6.9K bytes - Viewed (0)