Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TF_OperationGetAttrTensorShapeProto (0.29 sec)

  1. tensorflow/c/c_api.h

    // Sets `value` to the binary-serialized TensorShapeProto of the value of
    // `attr_name` attribute of `oper`.
    TF_CAPI_EXPORT extern void TF_OperationGetAttrTensorShapeProto(
        TF_Operation* oper, const char* attr_name, TF_Buffer* value,
        TF_Status* status);
    
    // Fills in `values` with binary-serialized TensorShapeProto values of the
    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", -1, TF_ATTR_SHAPE, 4);
      TF_Buffer* value = TF_NewBuffer();
      TF_OperationGetAttrTensorShapeProto(oper, "v", value, s_);
      EXPECT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      EXPECT_EQ(bytes.length(), value->length);
      EXPECT_EQ(0, memcmp(bytes.data(), value->data, value->length));
    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

          return;
        }
        storage_left -= n;
        for (int j = 0; j < n; ++j, ++p) {
          *p = shapes[i].dim_size(j);
        }
      }
    }
    
    void TF_OperationGetAttrTensorShapeProto(TF_Operation* oper,
                                             const char* attr_name,
                                             TF_Buffer* value, TF_Status* status) {
    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