- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for gradients (0.16 sec)
-
tensorflow/c/eager/gradients.cc
// Looks up the ID of a Gradient. int64_t TensorId(AbstractTensorHandle* tensor) const override; // Converts a Gradient to a TapeTensor. TapeTensor TapeTensorFromGradient(AbstractTensorHandle* g) const override; void MarkAsResult(AbstractTensorHandle* gradient) const override; void DeleteGradient(AbstractTensorHandle* gradient) const override; private:
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Sat Oct 12 05:11:17 UTC 2024 - 19.7K bytes - Viewed (0) -
tensorflow/c/eager/BUILD
":c_api_unified_internal", "//tensorflow/c/experimental/gradients/tape:tape_operation", "//tensorflow/core/lib/llvm_rtti", "//tensorflow/core/platform:errors", ], ) cc_library( name = "gradients_internal", srcs = [ "gradients.cc", ], hdrs = [ "gradients.h", "gradients_internal.h", ], visibility = [
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 11 23:52:39 UTC 2024 - 33.3K bytes - Viewed (0) -
tensorflow/c/eager/parallel_device/parallel_device.cc
// (where the value starts on the host), without allowing other implicit // copies/broadcasts. Other implicit copies may be supported eventually, // but need special handling for gradients (gradient of copy-on is not // just copy-off but includes a sum) and consideration of performance. // // TODO(allenl): There may be smarter ways to do this copy in some
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Mon Oct 21 04:14:14 UTC 2024 - 18.3K bytes - Viewed (0) -
tensorflow/c/eager/c_api.h
TF_Status* status); // Some TF ops need a step container to be set to limit the lifetime of some // resources (mostly TensorArray and Stack, used in while loop gradients in // graph mode). Calling this on a context tells it to start a step. TF_CAPI_EXPORT extern void TFE_ContextStartStep(TFE_Context* ctx); // Ends a step. When there is no active step (that is, every started step has
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Thu Apr 27 21:07:00 UTC 2023 - 22.8K bytes - Viewed (0) -
tensorflow/c/c_test_util.cc
} std::vector<std::pair<string, string>> GetGradDefs( const tensorflow::GraphDef& graph_def) { std::vector<std::pair<string, string>> grads; for (const tensorflow::GradientDef& grad : graph_def.library().gradient()) { grads.emplace_back(grad.function_name(), grad.gradient_func()); } std::sort(grads.begin(), grads.end()); return grads; }
Registered: Tue Nov 05 12:39:12 UTC 2024 - Last Modified: Fri Oct 15 03:16:52 UTC 2021 - 17.8K bytes - Viewed (0)