Searched refs:jacobian_ns (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker.cc | 340 std::vector<Tensor> jacobian_ns; in ComputeGradientErrorInternal() local
341 InitJacobians<X_T, Y_T, JAC_T>(xs, x_shapes, y_shapes, &jacobian_ns); in ComputeGradientErrorInternal()
345 scope, xs, x_shapes, ys, y_shapes, JAC_T{1e-3f}, x_datas, &jacobian_ns))); in ComputeGradientErrorInternal()
351 auto jac_n = jacobian_ns[i].matrix<JAC_T>(); in ComputeGradientErrorInternal()
|