- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for AbstractOperation (0.3 sec)
-
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) -
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 30 12:39:09 GMT 2024 - Last Modified: Wed Jul 14 16:20:41 GMT 2021 - 6.8K bytes - Viewed (0) -
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) -
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 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:27:35 GMT 2021 - 4.2K bytes - Viewed (0) -
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 30 12:39:09 GMT 2024 - Last Modified: Sun Oct 24 11:16:58 GMT 2021 - 3K bytes - Viewed (0) -
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) -
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 30 12:39:09 GMT 2024 - Last Modified: Fri Nov 13 22:20:40 GMT 2020 - 5.2K bytes - Viewed (0) -
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) -
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)