Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 8 of 8 for abstract_operation (0.16 sec)

  1. 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_; }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 7.3K bytes
    - Viewed (0)
  2. 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 {
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 3.6K bytes
    - Viewed (0)
  3. tensorflow/c/eager/abstract_context.h

    #define TENSORFLOW_C_EAGER_ABSTRACT_CONTEXT_H_
    
    #include <memory>
    
    #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).
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 3K bytes
    - Viewed (0)
  4. 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
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 5.2K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients_internal.h

    absl::Status SetAttrInt(AbstractOperation*, const char* attr_name,
                            int64_t value, ForwardOperation*);
    absl::Status SetAttrFloat(AbstractOperation*, const char* attr_name,
                              float value, ForwardOperation*);
    absl::Status SetAttrBool(AbstractOperation*, const char* attr_name, bool value,
                             ForwardOperation*);
    absl::Status SetAttrType(AbstractOperation*, const char* attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 4.7K bytes
    - Viewed (0)
  6. tensorflow/c/eager/gradients.cc

    }
    absl::Status SetAttrFunction(AbstractOperation* op_, const char* attr_name,
                                 const AbstractOperation* value,
                                 ForwardOperation* forward_op_) {
      return tensorflow::errors::Unimplemented(
          "SetAttrFunction has not been implemented yet.");
    }
    absl::Status SetAttrFunctionName(AbstractOperation* op_, const char* attr_name,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 19.7K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      }
      absl::Status SetAttrFunctionList(
          const char* attr_name,
          absl::Span<const AbstractOperation*> values) override {
        return tensorflow::errors::Unimplemented(
            "SetAttrFunctionList has not been implemented yet.");
      }
      // For LLVM style RTTI.
      static bool classof(const AbstractOperation* ptr) {
        return ptr->getKind() == kGraph;
      }
      ~GraphOperation() override {}
    
     private:
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 15.7K bytes
    - Viewed (0)
  8. tensorflow/c/eager/c_api.cc

                                   const TFE_Op** value, int num_values) {
      auto s = tensorflow::unwrap(op)->SetAttrFunctionList(
          attr_name, {reinterpret_cast<const tensorflow::AbstractOperation**>(
                          tensorflow::unwrap(value)),
                      static_cast<size_t>(num_values)});
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 05:11:17 UTC 2024
    - 43.9K bytes
    - Viewed (0)
Back to top