Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for dim (0.13 sec)

  1. tensorflow/c/c_api_test.cc

    #undef EXPECT_TF_META
    
    TEST(CAPI, TestTensorAligned) {
      int64_t dim = 7;
      size_t tensor_size_bytes = dim * TF_DataTypeSize(TF_FLOAT);
      TF_Tensor* a = TF_AllocateTensor(
          /*dtype=*/TF_FLOAT, /*dims=*/&dim, /*num_dims=*/1,
          /*len=*/tensor_size_bytes);
      float* data = reinterpret_cast<float*>(TF_TensorData(a));
      for (int i = 0; i < dim; ++i) {
        data[i] = 0;
      }
      if (TF_TensorDefaultAlignment() > 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)
  2. tensorflow/c/eager/c_api_test.cc

      ASSERT_EQ(num_dims, -1);
      ASSERT_EQ("Invalid handle", string(TF_Message(status.get())));
    
      TF_SetStatus(status.get(), TF_OK, "");
    
      int dim = TFE_TensorHandleDim(h, 0, status.get());
      ASSERT_EQ(TF_INVALID_ARGUMENT, TF_GetCode(status.get()));
      ASSERT_EQ(dim, -1);
      ASSERT_EQ("Invalid handle", string(TF_Message(status.get())));
    }
    
    TEST(CAPI, TensorHandleDevices) {
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Aug 03 20:50:20 GMT 2023
    - 94.6K bytes
    - Viewed (1)
Back to top