/aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
H A D | gradient_checker_test.cc | 47 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in TEST() 58 TF_ASSERT_OK((ComputeGradientError<double, double, double>( in TEST() 69 TF_ASSERT_OK((ComputeGradientError<complex64, complex64, float>( in TEST() 80 TF_ASSERT_OK((ComputeGradientError<complex128, complex128, double>( in TEST() 92 TF_ASSERT_OK((ComputeGradientError<float, complex64, float>( in TEST() 104 TF_ASSERT_OK((ComputeGradientError<complex64, float, float>( in TEST() 118 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in TEST() 134 TF_ASSERT_OK((ComputeGradientError<double, double, double>( in TEST() 149 TF_ASSERT_OK((ComputeGradientError<double, double, double>( in TEST() 164 TF_ASSERT_OK((ComputeGradientError<double, double, double>( in TEST() [all …]
|
H A D | gradient_checker.h | 49 Status ComputeGradientError(const Scope& scope, const OutputList& xs, 57 Status ComputeGradientError(const Scope& scope, const Output& x,
|
H A D | gradient_checker.cc | 378 Status ComputeGradientError(const Scope& scope, const OutputList& xs, in ComputeGradientError() function 406 Status ComputeGradientError(const Scope& scope, const Output& x, in ComputeGradientError() function 417 template Status ComputeGradientError<X_T, Y_T, JAC_T>( \ 421 template Status ComputeGradientError<X_T, Y_T, JAC_T>( \
|
/aosp_15_r20/external/tensorflow/tensorflow/cc/gradients/ |
H A D | image_grad_test.cc | 120 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestResizeToSmallerAndAlign() 135 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestResizeToLargerAndAlign() 220 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestScaleAndTranslate() 333 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestCropAndResize()
|
H A D | linalg_grad_test.cc | 39 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 48 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 91 TF_ASSERT_OK((ComputeGradientError<complex64, complex64, float>( in TEST_F()
|
H A D | functional_grad_test.cc | 40 auto result = (ComputeGradientError<float, float, float>( in RunTest() 51 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest()
|
H A D | math_grad_test.cc | 223 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, float>(scope_, x, x_data, y, in TestCWiseGrad() 626 TF_ASSERT_OK((ComputeGradientError<T, T, float>( in TestMatMulGrad() 748 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 757 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 1011 TF_ASSERT_OK((ComputeGradientError<double, float, double>( in TEST_F()
|
H A D | nn_grad_test.cc | 62 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 70 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 79 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest()
|
H A D | manip_grad_test.cc | 35 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest()
|
H A D | data_flow_grad_test.cc | 41 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest()
|
H A D | array_grad_test.cc | 39 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest() 48 TF_ASSERT_OK((ComputeGradientError<float, float, float>( in RunTest()
|