- Sort Score
- Result 10 results
- Languages All
Results 1 - 4 of 4 for RecordOperation (0.23 sec)
-
tensorflow/c/eager/gradients.cc
AbstractTensorHandle* AggregateGradients( gtl::ArraySlice<AbstractTensorHandle*> gradient_tensors) const override; // Calls the passed-in backward function. // op_type is the op's name provided in RecordOperation. Status CallBackwardFunction( const string& op_type, GradientFunction* gradient_function, const std::vector<int64_t>& unneeded_gradients, gtl::ArraySlice<AbstractTensorHandle*> output_gradients,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 19.3K bytes - Viewed (0) -
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/experimental/gradients/custom_gradient_test.cc
AbstractTensorHandle* exp_output; TF_RETURN_IF_ERROR(ops::Exp(ctx, inputs[0], &exp_output, "Exp")); std::unique_ptr<GradientFunction> gradient_function( new PassThroughGradientFunction); tape.RecordOperation(inputs, {exp_output}, gradient_function.release()); TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets*/ {exp_output}, /*sources=*/inputs,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 4.8K bytes - Viewed (0) -
tensorflow/c/eager/gradients_test.cc
Tape tape(/*persistent=*/false); tape.Watch(inputs[0]); AbstractTensorHandle* neg_output; TF_RETURN_IF_ERROR(ops::Neg(ctx, inputs[0], &neg_output, "Neg")); tape.RecordOperation(inputs, {neg_output}, nullptr, "Neg"); return tape.ComputeGradient(ctx, /*targets=*/{neg_output}, /*sources=*/inputs,
C++ - Registered: Tue Apr 30 12:39:09 GMT 2024 - Last Modified: Thu Feb 15 09:49:45 GMT 2024 - 7K bytes - Viewed (0)