Home
last modified time | relevance | path

Searched refs:X_T (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/cc/framework/
Dgradient_checker.cc102 template <typename X_T, typename Y_T, typename JAC_T>
142 const int x_stride = JacobianStride<X_T>::value; in ComputeTheoreticalJacobianTranspose()
163 auto dx_flat = dxout[x_idx].flat<X_T>(); in ComputeTheoreticalJacobianTranspose()
165 SetJacobian<X_T, JAC_T>(&jacobian, r * x_stride, in ComputeTheoreticalJacobianTranspose()
204 template <typename X_T, typename Y_T, typename JAC_T>
217 const int x_stride = JacobianStride<X_T>::value; in ComputeNumericJacobianTranspose()
222 auto x_data_flat = (*x_datas)[x_idx].flat<X_T>(); in ComputeNumericJacobianTranspose()
231 for (X_T unit : BaseUnitsForType<X_T>::values()) { in ComputeNumericJacobianTranspose()
232 X_T x_delta = unit * X_T{delta}; in ComputeNumericJacobianTranspose()
234 X_T v = x_data_flat(r); in ComputeNumericJacobianTranspose()
[all …]
Dgradient_checker.h48 template <typename X_T, typename Y_T, typename JAC_T>
56 template <typename X_T, typename Y_T, typename JAC_T>
/external/tensorflow/tensorflow/cc/gradients/
Dimage_grad_test.cc110 template <typename X_T, typename Y_T, typename JAC_T>
115 Tensor x_data = MakeData<X_T>(x_shape); in TestResizeToSmallerAndAlign()
117 MakeOp<X_T>(op_type, x_data, {2, 3}, align_corners, half_pixel_centers, &x, 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>
130 Tensor x_data = MakeData<X_T>(x_shape); in TestResizeToLargerAndAlign()
132 MakeOp<X_T>(op_type, x_data, {4, 6}, align_corners, half_pixel_centers, &x, 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()
[all …]
Dmath_grad_test.cc95 template <typename X_T, typename Y_T>
96 void TestCWiseGrad(UnaryOpType op_type, const std::function<X_T(int)>& x_fn) { in TestCWiseGrad()
98 DataType x_type = DataTypeToEnum<X_T>::v(); in TestCWiseGrad()
102 auto x_data_flat = x_data.flat<X_T>(); in TestCWiseGrad()
206 TF_ASSERT_OK((ComputeGradientError<X_T, Y_T, float>(scope_, x, x_data, y, in TestCWiseGrad()