Searched refs:jacobian_ts (Results 1 – 1 of 1) sorted by relevance
| /external/tensorflow/tensorflow/cc/framework/ |
| D | gradient_checker.cc | 108 std::vector<Tensor>* jacobian_ts) { in ComputeTheoreticalJacobianTranspose() argument
162 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeTheoreticalJacobianTranspose()
211 std::vector<Tensor>* jacobian_ts) { in ComputeNumericJacobianTranspose() argument
251 auto jacobian = (*jacobian_ts)[x_idx * y_num + y_idx].matrix<JAC_T>(); in ComputeNumericJacobianTranspose()
332 std::vector<Tensor> jacobian_ts; in ComputeGradientErrorInternal() local
333 InitJacobians<X_T, Y_T, JAC_T>(xs, x_shapes, y_shapes, &jacobian_ts); in ComputeGradientErrorInternal()
337 scope, xs, x_shapes, *x_datas, ys, y_shapes, &jacobian_ts))); in ComputeGradientErrorInternal()
347 for (int i = 0; i < jacobian_ts.size(); i++) { in ComputeGradientErrorInternal()
350 auto jac_t = jacobian_ts[i].matrix<JAC_T>(); in ComputeGradientErrorInternal()
352 for (int r = 0; r < jacobian_ts[i].dim_size(0); ++r) { in ComputeGradientErrorInternal()
[all …]
|