- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for TapeContext (0.17 sec)
-
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) -
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)