Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TapeOperation (0.2 sec)

  1. 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)
  2. 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)
  3. 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)
  4. 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)
  5. 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)
Back to top