Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_OperationGetAttrTensorList (0.23 sec)

  1. tensorflow/c/c_api.h

    //
    // The caller takes ownership of all the non-null TF_Tensor* entries in `values`
    // (which can be deleted using TF_DeleteTensor(values[i])).
    TF_CAPI_EXPORT extern void TF_OperationGetAttrTensorList(TF_Operation* oper,
                                                             const char* attr_name,
                                                             TF_Tensor** values,
    C
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Oct 26 21:08:15 GMT 2023
    - 82.3K bytes
    - Viewed (3)
  2. 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_TENSOR, -1);
      TF_Tensor* values[2];
      TF_OperationGetAttrTensorList(oper, "v", &values[0], TF_ARRAYSIZE(values),
                                    s_);
      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
    
      const char* tensor_data[] = {&tensor1[0], &tensor2[0]};
    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)
  3. tensorflow/c/c_api.cc

      Tensor t;
      status->status = tensorflow::GetNodeAttr(oper->node.attrs(), attr_name, &t);
      if (!status->status.ok()) return;
      *value = TF_TensorFromTensor(t, &status->status);
    }
    
    void TF_OperationGetAttrTensorList(TF_Operation* oper, const char* attr_name,
                                       TF_Tensor** values, int max_values,
                                       TF_Status* status) {
      std::vector<Tensor> ts;
    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