Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for SetAttrShape (0.25 sec)

  1. tensorflow/c/eager/c_api_unified_experimental_graph.cc

              "op_type and op_name must be specified before specifying attrs.");
        }
        op_->node_builder.Attr(attr_name, value);
        return absl::OkStatus();
      }
      Status SetAttrShape(const char* attr_name, const int64_t* dims,
                          const int num_dims) override {
        PartialTensorShape shape;
        if (num_dims >= 0) {
          shape = PartialTensorShape(ArraySlice<int64_t>(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 15.4K bytes
    - Viewed (1)
  2. tensorflow/c/eager/gradients.cc

                       DataType value, ForwardOperation* forward_op_) {
      forward_op_->attrs.Set(attr_name, value);
      return op_->SetAttrType(attr_name, value);
    }
    Status SetAttrShape(AbstractOperation* op_, const char* attr_name,
                        const int64_t* dims, const int num_dims,
                        ForwardOperation* forward_op_) {
      if (num_dims > TensorShape::MaxDimensions()) {
    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)
  3. tensorflow/c/eager/c_api.cc

      }
    }
    
    void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims,
                            const int num_dims, TF_Status* out_status) {
      out_status->status =
          tensorflow::unwrap(op)->SetAttrShape(attr_name, dims, num_dims);
    }
    
    void TFE_OpSetAttrFunction(TFE_Op* op, const char* attr_name,
                               const TFE_Op* value) {
      auto s = tensorflow::unwrap(op)->SetAttrFunction(
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Tue Mar 12 20:00:09 GMT 2024
    - 43.9K bytes
    - Viewed (2)
  4. tensorflow/c/c_api_test.cc

      EXPECT_TRUE(std::equal(std::begin(list), std::end(list), std::begin(values)));
    }
    
    TEST_F(CApiAttributesTest, Shape) {
      // Unknown shape
      auto desc = init("shape");
      TF_SetAttrShape(desc, "v", nullptr, -1);
      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_TF_META("v", -1, TF_ATTR_SHAPE, -1);
    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/c_api.cc

                            const char* value, size_t length) {
      tensorflow::NameAttrList func_name;
      func_name.set_name(string(value, value + length));
      desc->node_builder.Attr(attr_name, func_name);
    }
    
    void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name,
                         const int64_t* dims, int num_dims) {
      PartialTensorShape shape;
      if (num_dims >= 0) {
        shape = PartialTensorShape(
    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