Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for AbstractOperation (0.17 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/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 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  3. 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 30 12:39:09 GMT 2024
    - Last Modified: Mon Sep 26 22:40:32 GMT 2022
    - 3.6K bytes
    - Viewed (0)
  4. 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 30 12:39:09 GMT 2024
    - Last Modified: Wed Aug 31 17:39:44 GMT 2022
    - 1016 bytes
    - Viewed (0)
  5. 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)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

      }
      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:
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. 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;
      }
    }
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
Back to top