Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for AbstractOperation (0.28 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    namespace tensorflow {
    namespace gradients {
    TapeOperation::TapeOperation(AbstractOperation* parent_op, Tape* tape,
                                 const GradientRegistry& registry)
        : AbstractOperation(kTape),
          parent_op_(parent_op),
          tape_(tape),
          registry_(registry) {
      // TODO(b/172003047): Consider making AbstractOperation RefCounted.
      // parent_op_->Ref();
    }
    void TapeOperation::Release() {
    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/eager/tracing_utils.cc

    #include "tensorflow/core/lib/llvm_rtti/llvm_rtti.h"
    #include "tensorflow/core/platform/errors.h"
    
    namespace tensorflow {
    namespace tracing {
    
    Status MaybeSetOpName(AbstractOperation* op, const char* op_name) {
      if (isa<TracingOperation>(op)) {
        TF_RETURN_IF_ERROR(dyn_cast<TracingOperation>(op)->SetOpName(op_name));
      }
      if (isa<gradients::TapeOperation>(op)) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
Back to top