- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for GradientFunctionFactory (0.23 sec)
-
tensorflow/c/eager/gradients.h
std::vector<int64_t> skip_input_indices; AttrBuilder attrs; }; using GradientFunctionFactory = std::function<GradientFunction*(const ForwardOperation& op)>; // Map from op name to a `GradientFunctionFactory`. class GradientRegistry { public: Status Register(const string& op, GradientFunctionFactory gradient_function_factory); Status Lookup(const ForwardOperation& op,
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) -
tensorflow/c/eager/gradients.cc
op->Execute(absl::Span<AbstractTensorHandle*>(outputs), &num_outputs)); *result = outputs[0]; return absl::OkStatus(); } } // namespace Status GradientRegistry::Register( const string& op_name, GradientFunctionFactory gradient_function_factory) { auto iter = registry_.find(op_name); if (iter != registry_.end()) { const string error_msg = "Gradient already exists for op: " + op_name + ".";
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)