Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 22 for Attrs (0.12 sec)

  1. tensorflow/c/eager/abstract_op_attrs.h

    limitations under the License.
    ==============================================================================*/
    #ifndef TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    #define TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_
    
    #include "absl/container/inlined_vector.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    #include "tensorflow/core/framework/types.pb.h"
    #include "tensorflow/core/platform/status.h"
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed May 26 22:20:27 GMT 2021
    - 2K bytes
    - Viewed (0)
  2. tensorflow/c/eager/tfe_op_attrs_internal.h

    limitations under the License.
    ==============================================================================*/
    #ifndef TENSORFLOW_C_EAGER_TFE_OP_ATTRS_INTERNAL_H_
    #define TENSORFLOW_C_EAGER_TFE_OP_ATTRS_INTERNAL_H_
    
    #include "tensorflow/c/conversion_macros.h"
    #include "tensorflow/c/eager/abstract_op_attrs.h"
    #include "tensorflow/c/tf_status.h"
    #include "tensorflow/core/framework/attr_value.pb.h"
    
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Feb 10 05:41:19 GMT 2021
    - 1.6K bytes
    - Viewed (0)
  3. tensorflow/c/experimental/gradients/tape/tape_operation.cc

      forward_op_.attrs.Set(attr_name, value);
      return parent_op_->SetAttrFloat(attr_name, value);
    }
    Status TapeOperation::SetAttrBool(const char* attr_name, bool value) {
      forward_op_.attrs.Set(attr_name, value);
      return parent_op_->SetAttrBool(attr_name, value);
    }
    Status TapeOperation::SetAttrType(const char* attr_name, DataType value) {
      forward_op_.attrs.Set(attr_name, value);
    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/experimental/gradients/nn_grad.cc

    };
    
    // TODO(vnvo2409): Add python test
    class BiasAddGradientFunction : public GradientFunction {
     public:
      explicit BiasAddGradientFunction(AttrBuilder f_attrs)
          : forward_attrs_(f_attrs) {}
    
      Status Compute(AbstractContext* ctx,
                     absl::Span<AbstractTensorHandle* const> grad_outputs,
                     absl::Span<AbstractTensorHandle*> grad_inputs) override {
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5.7K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

      }
      // TODO(b/166669239): This is needed to support AttrBuilder::Get for string
      // attributes. Number type attrs and DataType attrs work fine without this.
      // Consider getting rid of this and making the behavior between number types
      // and string consistent.
      forward_op_->attrs.BuildNodeDef();
      std::unique_ptr<GradientFunction> gradient_fn;
      TF_RETURN_IF_ERROR(registry.Lookup(*forward_op_, &gradient_fn));
    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)
  6. tensorflow/c/experimental/gradients/math_grad.cc

    };
    
    class MatMulGradientFunction : public GradientFunction {
     public:
      explicit MatMulGradientFunction(vector<AbstractTensorHandle*> f_inputs,
                                      AttrBuilder f_attrs)
          : forward_inputs_(f_inputs), forward_attrs_(f_attrs) {
        for (auto input : forward_inputs_) {
          if (input) {
            input->Ref();
          }
        }
      }
    
      Status Compute(AbstractContext* ctx,
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 15.2K bytes
    - Viewed (0)
  7. tensorflow/c/eager/c_api_test.cc

      tensorflow::NameAttrList name_and_attrs;
      ASSERT_TRUE(name_and_attrs.ParseFromArray(serialized_attr_values->data,
                                                serialized_attr_values->length));
      ASSERT_EQ("VarHandleOp", name_and_attrs.name());
      ASSERT_EQ(tensorflow::DT_INT64,
                name_and_attrs.attr().find("dtype")->second.type());
      TF_DeleteBuffer(serialized_attr_values);
    
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
  8. tensorflow/c/eager/c_api_internal.h

    #include "tensorflow/c/eager/tfe_executor_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_monitoring_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_op_attrs_internal.h"  // IWYU pragma: export
    #include "tensorflow/c/eager/tfe_tensor_debug_info_internal.h"  // IWYU pragma: export
    
    // TODO(b/154564140): Move this to its own header. This requires splitting
    // c_api_experimental.h
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Wed Jan 18 19:26:34 GMT 2023
    - 1.9K bytes
    - Viewed (0)
  9. tensorflow/c/eager/immediate_execution_operation.h

      virtual void SetStackTrace(ManagedStackTrace stack_trace) = 0;
    
      virtual const tensorflow::AbstractOpAttrs* GetOpAttrs() const = 0;
      virtual void AddAttrs(const AbstractOpAttrs* op_attrs) = 0;
    
      virtual void SetCancellationManager(
          CancellationManager* cancellation_manager) = 0;
    
      // Returns the stack trace set by `SetStackTrace` if exists.
    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)
  10. tensorflow/c/c_api.cc

    }
    
    int TF_OperationGetNumAttrs(TF_Operation* oper) {
      return oper->node.attrs().size();
    }
    
    int TF_OperationGetAttrNameLength(TF_Operation* oper, int i) {
      auto attrs = oper->node.attrs();
      int count = 0;
      AttrValueMap::const_iterator it;
      for (it = attrs.begin(); it != attrs.end(); it++) {
        if (count == i) {
          return it->first.length();
        }
        count++;
      }
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 102.3K bytes
    - Viewed (0)
Back to top