- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TapeOperation (0.2 sec)
-
tensorflow/c/experimental/gradients/tape/tape_operation.cc
delete this; } TapeOperation::~TapeOperation() { // TODO(b/172003047): Consider making AbstractOperation RefCounted. // parent_op->Unref(); } Status TapeOperation::Reset(const char* op, const char* raw_device_name) { forward_op_.op_name = op; forward_op_.attrs.Reset(op); forward_op_.inputs.clear(); forward_op_.outputs.clear(); return parent_op_->Reset(op, raw_device_name); }
C++ - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Tue Jun 07 01:53:35 GMT 2022 - 9K bytes - Viewed (1) -
tensorflow/c/experimental/gradients/tape/tape_operation.h
#include "tensorflow/c/eager/abstract_operation.h" #include "tensorflow/c/eager/gradients.h" namespace tensorflow { namespace gradients { class TapeOperation : public AbstractOperation { public: explicit TapeOperation(AbstractOperation*, Tape*, const GradientRegistry&); void Release() override; Status Reset(const char* op, const char* raw_device_name) override; const string& Name() const override;
C - Registered: Tue Feb 27 12:39:08 GMT 2024 - Last Modified: Wed Sep 23 23:12:39 GMT 2020 - 3.7K bytes - Viewed (1) -
tensorflow/c/eager/tracing_utils.cc
if (isa<TracingOperation>(op)) { TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name)); } if (isa<gradients::TapeOperation>(op)) { TF_RETURN_IF_ERROR(MaybeSetOpName( dyn_cast<gradients::TapeOperation>(op)->GetBackingOperation(), op_name)); } return absl::OkStatus(); } } // namespace tracing
C++ - Registered: Tue Apr 23 12:39:09 GMT 2024 - Last Modified: Tue Feb 27 13:57:45 GMT 2024 - 1.4K bytes - Viewed (0) -
tensorflow/c/experimental/gradients/tape/tape_context.cc
// TODO(srbs): Change to Unref() delete this; } 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); }
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
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; Status RemoveFunction(const string& func) override; // For LLVM style RTTI. static bool classof(const AbstractContext* ptr) {
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)