Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for tensor_ndims (0.1 sec)

  1. tensorflow/c/c_api_test.cc

      const size_t tensor_size[] = {TF_ARRAYSIZE(tensor1), TF_ARRAYSIZE(tensor2)};
      const int64_t* tensor_dims[] = {&dims1[0], &dims2[0]};
      const size_t tensor_ndims[] = {ndims1, ndims2};
      for (int i = 0; i < 2; ++i) {
        TF_Tensor* v = values[i];
        ASSERT_NE(nullptr, v) << i;
        EXPECT_EQ(TF_INT8, TF_TensorType(v)) << i;
        EXPECT_EQ(tensor_ndims[i], TF_NumDims(v)) << i;
        for (int j = 0; j < TF_NumDims(v); ++j) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Apr 15 03:35:10 UTC 2024
    - 96.9K bytes
    - Viewed (0)
Back to top