Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TF_NumDims (0.24 sec)

  1. tensorflow/c/eager/gradient_checker.cc

      for (int32_t i = start; i < end; i += step) {
        (*data)[i] = i;
      }
    }
    
    // Fills out_dims with the dimensions of the given tensor.
    void GetDims(const TF_Tensor* t, int64_t* out_dims) {
      int num_dims = TF_NumDims(t);
      for (int i = 0; i < num_dims; i++) {
        out_dims[i] = TF_Dim(t, i);
      }
    }
    
    // Runs model as is if output is a scalar,
    // else sums the output tensor before returning.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Feb 15 09:49:45 GMT 2024
    - 7.3K bytes
    - Viewed (0)
  2. tensorflow/c/experimental/gradients/grad_test_helper.cc

      TF_Tensor* analytical_tensor;
      auto s = GetValue(t, &analytical_tensor);
      ASSERT_EQ(errors::OK, s.code()) << s.message();
    
      int64_t num_elem_analytical = 1;
      auto num_dims_analytical = TF_NumDims(analytical_tensor);
      ASSERT_EQ(dims.size(), num_dims_analytical);
      for (int j = 0; j < num_dims_analytical; j++) {
        auto dim_analytical = TF_Dim(analytical_tensor, j);
        ASSERT_EQ(dims[j], dim_analytical);
    C++
    - Registered: Tue Mar 26 12:39:09 GMT 2024
    - Last Modified: Wed Feb 28 13:53:47 GMT 2024
    - 5K bytes
    - Viewed (0)
  3. 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));
    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)
  4. 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.
    C++
    - Registered: Tue Apr 30 12:39:09 GMT 2024
    - Last Modified: Thu Jul 20 22:08:54 GMT 2023
    - 63.6K bytes
    - Viewed (6)
Back to top