- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for TF_Dim (0.19 sec)
-
tensorflow/c/experimental/gradients/grad_test_helper.cc
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); num_elem_analytical *= dim_analytical; } float* danalytical = new float[num_elem_analytical]{0};
C++ - Registered: Tue Mar 26 12:39:09 GMT 2024 - Last Modified: Wed Feb 28 13:53:47 GMT 2024 - 5K bytes - Viewed (0) -
tensorflow/c/eager/gradient_checker.cc
} } // 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. Status RunAndMaybeSum(AbstractContext* ctx, Model forward,
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) -
tensorflow/c/c_api_test.cc
&Deallocator, &deallocator_called); EXPECT_FALSE(deallocator_called); EXPECT_EQ(TF_FLOAT, TF_TensorType(t)); EXPECT_EQ(2, TF_NumDims(t)); EXPECT_EQ(dims[0], TF_Dim(t, 0)); EXPECT_EQ(dims[1], TF_Dim(t, 1)); EXPECT_EQ(num_bytes, TF_TensorByteSize(t)); EXPECT_EQ(static_cast<void*>(values), TF_TensorData(t)); TF_DeleteTensor(t); EXPECT_TRUE(deallocator_called); }
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)