Home
last modified time | relevance | path

Searched refs:JAC_T (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc66 template <typename T, typename JAC_T>
68 typename TTypes<JAC_T>::Matrix* jacobian, const int row, const int col, in SetJacobian()
70 (*jacobian)(row, col) = JAC_T{value}; in SetJacobian()
73 template <typename T, typename JAC_T>
75 typename TTypes<JAC_T>::Matrix* jacobian, const int row, const int col, in SetJacobian()
77 (*jacobian)(row, col) = JAC_T{value.real()}; in SetJacobian()
79 (*jacobian)(row + 1, col) = JAC_T{value.imag()}; in SetJacobian()
81 (*jacobian)(row, col + 1) = JAC_T{value.imag()}; in SetJacobian()
102 template <typename X_T, typename Y_T, typename JAC_T>
162 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeTheoreticalJacobianTranspose()
[all …]
Dgradient_checker.h48 template <typename X_T, typename Y_T, typename JAC_T>
53 JAC_T* max_error);
56 template <typename X_T, typename Y_T, typename JAC_T>
59 const TensorShape& y_shape, JAC_T* max_error);
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc110 template <typename X_T, typename Y_T, typename JAC_T>
119 JAC_T max_error; in TestResizeToSmallerAndAlign()
120 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestResizeToSmallerAndAlign()
125 template <typename X_T, typename Y_T, typename JAC_T>
134 JAC_T max_error; in TestResizeToLargerAndAlign()
135 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, JAC_T>( in TestResizeToLargerAndAlign()
140 template <typename X_T, typename Y_T, typename JAC_T>
148 TestResizeToSmallerAndAlign<X_T, Y_T, JAC_T>(op_type, align_corners, in TestResize()
150 TestResizeToLargerAndAlign<X_T, Y_T, JAC_T>(op_type, align_corners, in TestResize()
210 template <typename X_T, typename Y_T, typename JAC_T>
[all …]