Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for TF_NumDims (0.04 sec)

  1. tensorflow/c/c_api_test.cc

      ASSERT_EQ(TF_OK, TF_GetCode(s_)) << TF_Message(s_);
      ASSERT_NE(nullptr, value);
      EXPECT_EQ(TF_INT8, TF_TensorType(value));
      EXPECT_EQ(ndims, TF_NumDims(value));
      for (int i = 0; i < TF_NumDims(value); ++i) {
        EXPECT_EQ(dims[i], TF_Dim(value, i)) << i;
      }
      EXPECT_EQ(sizeof(char) * TF_ARRAYSIZE(tensor), TF_TensorByteSize(value));
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Sat Oct 12 16:27:48 UTC 2024
    - 97K bytes
    - Viewed (0)
  2. tensorflow/c/c_api_function_test.cc

        for (int i = 0; i < expected_results.size(); ++i) {
          TF_Tensor* out = csession.output_tensor(i);
          ASSERT_TRUE(out != nullptr);
          EXPECT_EQ(TF_INT32, TF_TensorType(out));
          EXPECT_EQ(1, TF_NumDims(out));
          CompareInt32Tensor(expected_results[i], out);
        }
      }
    
      // Run the host graph, which now contains a function and check that
      // outputs are as expected.
    Registered: Tue Nov 05 12:39:12 UTC 2024
    - Last Modified: Thu Jul 20 22:08:54 UTC 2023
    - 63.6K bytes
    - Viewed (0)
Back to top