Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 37 for attr (0.19 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/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)
  6. tensorflow/c/c_api.cc

      desc->node_builder.Attr(
          attr_name, ArraySlice<const DataType>(
                         reinterpret_cast<const DataType*>(values), num_values));
    }
    
    void TF_SetAttrPlaceholder(TF_OperationDescription* desc, const char* attr_name,
                               const char* placeholder) {
      tensorflow::AttrValue attr_value;
      attr_value.set_placeholder(placeholder);
      desc->node_builder.Attr(attr_name, attr_value);
    }
    
    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)
  7. 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)
  8. tensorflow/c/c_test_util.cc

      bool found_value = false;
      for (const auto& attr : node_def.attr()) {
        if (attr.first == "dtype") {
          if (attr.second.type() == tensorflow::DT_INT32) {
            found_dtype = true;
          } else {
            return false;
          }
        } else if (attr.first == "value") {
          if (attr.second.has_tensor() &&
              attr.second.tensor().int_val_size() == 1 &&
              attr.second.tensor().int_val(0) == v) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Fri Oct 15 03:16:52 GMT 2021
    - 17.8K bytes
    - Viewed (2)
  9. 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)
  10. tensorflow/c/experimental/gradients/tape/tape_operation.h

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