Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 10 of 11 for SetAttrIntList (0.21 sec)

  1. tensorflow/c/experimental/gradients/tape/tape_operation.h

                               const size_t* lengths, int num_values) override;
      Status SetAttrFloatList(const char* attr_name, const float* values,
                              int num_values) override;
      Status SetAttrIntList(const char* attr_name, const int64_t* values,
                            int num_values) override;
      Status SetAttrTypeList(const char* attr_name, const DataType* values,
                             int num_values) 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)
  2. tensorflow/c/experimental/gradients/tape/tape_operation.cc

    }
    Status TapeOperation::SetAttrIntList(const char* attr_name,
                                         const int64_t* values, int num_values) {
      forward_op_.attrs.Set(
          attr_name, gtl::ArraySlice<const int64_t>(
                         reinterpret_cast<const int64_t*>(values), num_values));
      return parent_op_->SetAttrIntList(attr_name, values, num_values);
    }
    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/gradients_internal.h

                             int num_values, ForwardOperation*);
    Status SetAttrFloatList(AbstractOperation*, const char* attr_name,
                            const float* values, int num_values, ForwardOperation*);
    Status SetAttrIntList(AbstractOperation*, const char* attr_name,
                          const int64_t* values, int num_values, ForwardOperation*);
    Status SetAttrTypeList(AbstractOperation*, const char* attr_name,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Sun Oct 24 11:27:35 GMT 2021
    - 4.2K bytes
    - Viewed (0)
  4. tensorflow/c/eager/abstract_operation.h

                                       absl::Span<string const> values);
      virtual Status SetAttrFloatList(const char* attr_name, const float* values,
                                      int num_values) = 0;
      virtual Status SetAttrIntList(const char* attr_name, const int64_t* values,
                                    int num_values) = 0;
      virtual Status SetAttrTypeList(const char* attr_name, const DataType* values,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Wed Jul 14 16:20:41 GMT 2021
    - 6.8K bytes
    - Viewed (0)
  5. tensorflow/c/eager/gradients.cc

    }
    Status SetAttrIntList(AbstractOperation* op_, const char* attr_name,
                          const int64_t* values, int num_values,
                          ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(
          attr_name, gtl::ArraySlice<const int64_t>(
                         reinterpret_cast<const int64_t*>(values), num_values));
      return op_->SetAttrIntList(attr_name, values, num_values);
    }
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 19.3K bytes
    - Viewed (0)
  6. tensorflow/c/eager/c_api_unified_experimental_graph.cc

                              int num_values) override {
        op_->node_builder.Attr(attr_name,
                               ArraySlice<const float>(values, num_values));
        return absl::OkStatus();
      }
      Status SetAttrIntList(const char* attr_name, const int64_t* values,
                            int num_values) override {
        op_->node_builder.Attr(
            attr_name, ArraySlice<const int64_t>(
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  7. tensorflow/c/eager/parallel_device/parallel_device_test.cc

                       merge_op.length());
      const std::string final_op("Id");
      TF_SetAttrString(reduce_desc, "final_op", final_op.c_str(),
                       final_op.length());
      TF_SetAttrIntList(reduce_desc, "subdiv_offsets", nullptr, 0);
      TF_AddInput(reduce_desc, x);
      TF_Operation* reduce_op = TF_FinishOperation(reduce_desc, status);
      if (TF_GetCode(status) != TF_OK) return;
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Jul 08 23:47:35 GMT 2021
    - 29.3K bytes
    - Viewed (1)
  8. tensorflow/c/c_api_test.cc

      EXPECT_EQ(31415, value);
    }
    
    TEST_F(CApiAttributesTest, IntList) {
      const int64_t list[] = {1, 2, 3, 4};
      const size_t list_size = TF_ARRAYSIZE(list);
    
      auto desc = init("list(int)");
      TF_SetAttrIntList(desc, "v", list, list_size);
    
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      int64_t values[list_size];
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Mon Apr 15 03:35:10 GMT 2024
    - 96.9K bytes
    - Viewed (3)
  9. tensorflow/c/eager/c_api.cc

      }
    }
    
    void TFE_OpSetAttrIntList(TFE_Op* op, const char* attr_name,
                              const int64_t* values, int num_values) {
      auto s =
          tensorflow::unwrap(op)->SetAttrIntList(attr_name, values, num_values);
      if (!s.ok()) {
        LOG(WARNING) << "Unable to set attribute: " << attr_name;
      }
    }
    
    void TFE_OpSetAttrTypeList(TFE_Op* op, const char* attr_name,
    C++
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  10. tensorflow/c/c_api.h

    TF_CAPI_EXPORT extern void TF_SetAttrInt(TF_OperationDescription* desc,
                                             const char* attr_name, int64_t value);
    TF_CAPI_EXPORT extern void TF_SetAttrIntList(TF_OperationDescription* desc,
                                                 const char* attr_name,
                                                 const int64_t* values,
    C
    - Registered: Tue Apr 23 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
Back to top