Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 12 for AbstractOperation (0.31 sec)

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

      Status SetAttrFunctionList(
          const char* attr_name,
          absl::Span<const AbstractOperation*> values) override;
      AbstractOperation* GetBackingOperation();
      // For LLVM style RTTI.
      static bool classof(const AbstractOperation* ptr) {
        return ptr->getKind() == kTape;
      }
      ~TapeOperation() override;
    
     private:
      AbstractOperation* parent_op_;
      ForwardOperation forward_op_;
      Tape* tape_;
    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)
  2. tensorflow/c/eager/abstract_operation.h

    // tracing or immediate execution mode.
    class AbstractOperation {
     protected:
      enum AbstractOperationKind {
        kGraph,
        kMlir,
        kEager,
        kTfrt,
        kTape,
        kOpHandler
      };
      explicit AbstractOperation(AbstractOperationKind kind) : kind_(kind) {}
      virtual ~AbstractOperation() {}
    
     public:
      AbstractOperationKind getKind() const { return kind_; }
    
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  3. 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)
  4. tensorflow/c/eager/gradients.cc

    }
    Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
                           const AbstractOperation* value,
                           ForwardOperation* forward_op_) {
      return tensorflow::errors::Unimplemented(
          "SetAttrFunction has not been implemented yet.");
    }
    Status SetAttrFunctionName(AbstractOperation* op_, const char* attr_name,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  5. tensorflow/c/eager/abstract_context.h

    #include "tensorflow/c/eager/abstract_function.h"
    #include "tensorflow/c/eager/abstract_operation.h"
    
    namespace tensorflow {
    
    // Abstract interface to a context.
    //
    // This serves as a factory for creating `AbstractOperation`s and for
    // registering traced functions.
    // Operations creation within a context can only be executed in that context
    // (for now at least).
    // Implementations of the context may contain some state e.g. an execution
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:16:58 GMT 2021
    - 3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients_internal.h

    Status SetAttrFloat(AbstractOperation*, const char* attr_name, float value,
                        ForwardOperation*);
    Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value,
                       ForwardOperation*);
    Status SetAttrType(AbstractOperation*, const char* attr_name, DataType value,
                       ForwardOperation*);
    Status SetAttrShape(AbstractOperation*, const char* attr_name,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  7. tensorflow/c/eager/immediate_execution_operation.h

      virtual void SetStepId(int64_t step_id) = 0;
    
      // For LLVM style RTTI.
      static bool classof(const AbstractOperation* ptr) {
        return ptr->getKind() == kEager || ptr->getKind() == kTfrt;
      }
    
     protected:
      explicit ImmediateExecutionOperation(AbstractOperationKind kind)
          : AbstractOperation(kind) {}
      ~ImmediateExecutionOperation() override {}
    };
    
    namespace internal {
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Sep 26 22:40:32 GMT 2022
    - 3.6K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api_unified_experimental_internal.h

    // It is allowed to reusing the same abstract operation for multiple execution
    // on a given context, with the same or different input tensors.
    class TracingOperation : public AbstractOperation {
     protected:
      explicit TracingOperation(AbstractOperationKind kind)
          : AbstractOperation(kind) {}
    
     public:
      // Sets the name of the operation: this is an optional identifier that is
      // not intended to carry semantics and preserved/propagated without
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Fri Nov 13 22:20:40 GMT 2020
    - 5.2K bytes
    - Viewed (0)
  9. tensorflow/c/eager/tracing_utils.h

    #ifndef TENSORFLOW_C_EAGER_TRACING_UTILS_H_
    #define TENSORFLOW_C_EAGER_TRACING_UTILS_H_
    
    #include "tensorflow/c/eager/abstract_operation.h"
    
    namespace tensorflow {
    namespace tracing {
    Status MaybeSetOpName(AbstractOperation*, const char* op_name);
    }  // namespace tracing
    }  // namespace tensorflow
    
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Aug 31 17:39:44 GMT 2022
    - 1016 bytes
    - Viewed (0)
  10. 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 23 12:39:09 GMT 2024
    - Last Modified: Tue Feb 27 13:57:45 GMT 2024
    - 1.4K bytes
    - Viewed (0)
Back to top