Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_OperationGetAttrTensorShapeProtoList (0.21 sec)

  1. tensorflow/c/c_api.h

    // least `num_values` (ideally set to TF_AttrMetadata.list_size from
    // TF_OperationGetAttrMetadata(oper, attr_name)).
    TF_CAPI_EXPORT extern void TF_OperationGetAttrTensorShapeProtoList(
        TF_Operation* oper, const char* attr_name, TF_Buffer** values,
        int max_values, TF_Status* status);
    
    // Gets the TF_Tensor valued attribute of `attr_name` of `oper`.
    //
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Oct 26 21:08:15 UTC 2023
    - 82.3K bytes
    - Viewed (0)
  2. tensorflow/c/c_api.cc

        status->status =
            InvalidArgument("Value for '", attr_name, "' is not a shape.");
        return;
      }
      status->status = MessageToBuffer(attr->shape(), value);
    }
    
    void TF_OperationGetAttrTensorShapeProtoList(TF_Operation* oper,
                                                 const char* attr_name,
                                                 TF_Buffer** values, int max_values,
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 102.3K bytes
    - Viewed (0)
  3. tensorflow/c/c_api_test.cc

      auto oper = TF_FinishOperation(desc, s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      EXPECT_TF_META("v", 2, TF_ATTR_SHAPE, 8);
      TF_Buffer* values[2];
      TF_OperationGetAttrTensorShapeProtoList(oper, "v", values, 2, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      for (int i = 0; i < 2; ++i) {
        int le = list_lens[i];
        int la = values[i]->length;
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
Back to top