Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 38 for attrId (0.2 sec)

  1. tensorflow/c/eager/abstract_op_attrs.h

      virtual void GetNameAttrList(
          tensorflow::NameAttrList* name_and_attrs) const = 0;
    
      virtual bool GetInt(absl::string_view, int64_t* result) const = 0;
      virtual bool GetFloat(absl::string_view attr_name, float* result) const = 0;
      virtual bool GetBool(absl::string_view attr_name, bool* result) const = 0;
      virtual bool GetType(absl::string_view attr_name, DataType* result) const = 0;
      virtual Status GetTypeList(
    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/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)
  3. tensorflow/c/eager/c_api_test.cc

      tensorflow::AttrValueMap attr_values = ExtractAttrs(minOp);
      tensorflow::AttrValueMap::const_iterator attr_found = attr_values.find("T");
      EXPECT_NE(attr_found, attr_values.cend());
      EXPECT_EQ(attr_found->second.type(), tensorflow::DataType::DT_FLOAT);
      attr_found = attr_values.find("Tidx");
      EXPECT_NE(attr_found, attr_values.cend());
      EXPECT_EQ(attr_found->second.type(), tensorflow::DataType::DT_INT32);
    
    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)
  4. tensorflow/c/c_api_test.cc

    ATTR_TEST_REGISTER_OP(string);
    ATTR_TEST_REGISTER_OP(int);
    ATTR_TEST_REGISTER_OP(float);
    ATTR_TEST_REGISTER_OP(bool);
    ATTR_TEST_REGISTER_OP(type);
    ATTR_TEST_REGISTER_OP(shape);
    ATTR_TEST_REGISTER_OP(tensor);
    #undef ATTR_TEST_REGISTER_OP
    
    class CApiAttributesTest : public ::testing::Test {
     protected:
      CApiAttributesTest()
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  5. tensorflow/c/eager/tfe_op_attrs_internal.h

    ==============================================================================*/
    #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"
    
    // An equivalent of a tensorflow::NameAttrList protocol buffer, but used in ways
    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)
  6. tensorflow/c/eager/gradients.cc

      return op_->SetAttrInt(attr_name, value);
    }
    Status SetAttrFloat(AbstractOperation* op_, const char* attr_name, float value,
                        ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrFloat(attr_name, value);
    }
    Status SetAttrBool(AbstractOperation* op_, const char* attr_name, bool value,
    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)
  7. tensorflow/c/c_api.cc

                                              const char* attr_name,
                                              TF_Status* status) {
      const tensorflow::AttrValue* attr = oper->node.attrs().Find(attr_name);
      if (attr == nullptr) {
        status->status = InvalidArgument("Operation '", oper->node.name(),
                                         "' has no attr named '", attr_name, "'.");
      }
      return attr;
    }
    
    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)
  8. tensorflow/c/eager/c_api_unified_experimental.h

    // TODO(srbs): Add APIs for specifying attrs etc.
    // `op_type` must outlive `op`.
    void TF_AbstractOpSetOpType(TF_AbstractOp* op, const char* const op_type,
                                TF_Status* s);
    // `op_name` must outlive `op`.
    void TF_AbstractOpSetOpName(TF_AbstractOp* op, const char* const op_name,
                                TF_Status* s);
    // `attr_name` must outlive `op`.
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:00 GMT 2021
    - 7K bytes
    - Viewed (0)
  9. 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)
  10. tensorflow/c/eager/abstract_operation.h

      virtual Status SetAttrString(const char* attr_name, const char* data,
                                   size_t length) = 0;
      virtual Status SetAttrInt(const char* attr_name, int64_t value) = 0;
      virtual Status SetAttrFloat(const char* attr_name, float value) = 0;
      virtual Status SetAttrBool(const char* attr_name, bool value) = 0;
      virtual Status SetAttrType(const char* attr_name, DataType value) = 0;
    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)
Back to top