- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TapeContext (0.18 sec)
-
tensorflow/c/experimental/gradients/tape/tape_context.cc
TapeContext::~TapeContext() { // TODO(srbs): Make AbstractContext ref counted. // parent_ctx_->Unref(); } TapeOperation* TapeContext::CreateOperation() { return new TapeOperation(parent_ctx_->CreateOperation(), tape_, registry_); } Status TapeContext::RegisterFunction(AbstractFunction* f) { return parent_ctx_->RegisterFunction(f); } Status TapeContext::RemoveFunction(const string& func) { return parent_ctx_->RemoveFunction(func); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.7K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_context.h
#include "tensorflow/c/eager/abstract_context.h" #include "tensorflow/c/experimental/gradients/tape/tape_operation.h" namespace tensorflow { namespace gradients { class TapeContext : public AbstractContext { public: explicit TapeContext(AbstractContext*, Tape*, const GradientRegistry&); void Release() override; TapeOperation* CreateOperation() override; Status RegisterFunction(AbstractFunction*) override;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 1.6K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/grad_test_helper.cc
Tape tape(/*persistent=*/false); for (size_t i{}; i < inputs.size(); ++i) { tape.Watch(inputs[i]); } std::vector<AbstractTensorHandle*> temp_outputs(1); AbstractContextPtr tape_ctx(new TapeContext(ctx, &tape, grad_registry)); TF_RETURN_IF_ERROR( forward_model(tape_ctx.get(), inputs, absl::MakeSpan(temp_outputs))); TF_RETURN_IF_ERROR(tape.ComputeGradient(ctx, /*targets=*/temp_outputs,
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0)