Home
last modified time | relevance | path

Searched refs:numerical_tensor (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/experimental/gradients/
H A Dgrad_test_helper.cc48 TF_Tensor* numerical_tensor; in CompareNumericalAndAutodiffGradients() local
49 s = GetValue(numerical_grad.get(), &numerical_tensor); in CompareNumericalAndAutodiffGradients()
51 auto num_elem_numerical = TF_TensorElementCount(numerical_tensor); in CompareNumericalAndAutodiffGradients()
61 memcpy(&dnumerical[0], TF_TensorData(numerical_tensor), in CompareNumericalAndAutodiffGradients()
62 TF_TensorByteSize(numerical_tensor)); in CompareNumericalAndAutodiffGradients()
71 TF_DeleteTensor(numerical_tensor); in CompareNumericalAndAutodiffGradients()
/aosp_15_r20/external/tensorflow/tensorflow/c/eager/
H A Dgradient_checker_test.cc48 TF_Tensor* numerical_tensor; in CompareNumericalAndManualGradients() local
49 s = GetValue(numerical_grad.get(), &numerical_tensor); in CompareNumericalAndManualGradients()
51 auto num_elem_numerical = TF_TensorElementCount(numerical_tensor); in CompareNumericalAndManualGradients()
55 memcpy(&dnumerical[0], TF_TensorData(numerical_tensor), in CompareNumericalAndManualGradients()
56 TF_TensorByteSize(numerical_tensor)); in CompareNumericalAndManualGradients()
62 TF_DeleteTensor(numerical_tensor); in CompareNumericalAndManualGradients()